<scp>Frugal:</scp> Efficient and Economic Embedding Model Training with Commodity GPUs
Embedding models show superiority in learning representations of massive ID-type features in sparse learning scenarios such as recommendation systems (e.g., user/item IDs) and graph learning (e.g., node/edge IDs). Commodity GPUs are highly favored for their cost-efficient computing power, which is ideally suited for the low computing demand of memory-intensive embedding models. However, directly running embedding model training on commodity GPUs yields poor performance because of their deficient communication resources (including low communication bandwidth and no PCIe P2P support).
Authors
- Youyou Lu (ORCID: https://orcid.org/0000-0002-6214-5390)
- Shaoxun Zeng (ORCID: https://orcid.org/0009-0004-6493-1768)
- Hao Guo (ORCID: https://orcid.org/0009-0002-4819-5786)
- Minhui Xie (ORCID: https://orcid.org/0000-0001-6684-8336)
- S.Y. Gao (ORCID: https://orcid.org/0009-0000-4751-0161)
Institutions
- Renmin University of China (CN)
- Tsinghua University (CN)
Publication Details
- Published
- 2025-02-06
- DOI
- https://doi.org/10.1145/3669940.3707245
- Primary Topic
- Recommender Systems and Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.00