边缘智能计算系统中加速推荐模型训练的样本调度机制.pdfVIP

计算机研究与发展

Journal

of

Computer

Research

and

Development62（6）：1396−1415，2025

边缘智能计算系统中加速推荐模型训练的样本调度机制

李国鹏谈海生张弛倪宏秋王子龙章馨月徐洋田晗陈国良121111111

1

（中国科学技术大学计算机科学与技术学院合肥230027）

2

（中国科学技术大学人工智能与数据科学学院合肥230027）

（guopengli@）

SamplesDispatchingMechanismforAcceleratingRecommendationModelTraining

inEdgeIntelligentComputingSystem

1

2

1

1

1

1

1

1

Li

Guopeng,

Tan

Haisheng,

Zhang

Chi,

Ni

Hongqiu,

Wang

Zilong,

Zhang

Xinyue,

Xu

Yang,

Tian

Han,

and

Chen

Guoliang1

1

（SchoolofComputerScienceandTechnology,UniversityofScienceandTechnologyofChina,Hefei230027）

2

（SchoolofArtificialIntelligenceandDataScience,UniversityofScienceandTechnologyofChina,Hefei230027）

AbstractUsing

edge

workers

to

train

deep

learning

recommendation

model

(DLRM)

in

edge

intelligent

computing

system

brings

several

benefits,

particularly

in

terms

of

data

privacy

protection,

low

latency

and

personalization.

However,

due

to

the

huge

size

of

embedding

tables,

typical

DLRM

training

frameworks

adopt

one

or

more

parameter

servers

to

maintain

global

embedding

tables,

while

leveraging

several

edge

workers

to

cache

part

of

them.

This

incurs

significant

transmission

cost

for

embedding

transmissions

between

workers

and

parameter

servers,

which

can

dominate

the

training

cycle.

In

this

paper,

we

investigate

how

to

dispatch

input

embedding

samples

to

appropriate

edge

workers

to

minimize

the

total

embedding

transmission

cost

when

facing

edge-specific

challenges

such

a