Advanced
Open
Pro
Sizing and Sharding a Large Embedding Table
An ads CTR model needs embeddings for 800 M user IDs, 50 M ad IDs and one user×ad-category cross with 2 B possible values. The plan is 32 dimensions in fp32 for every table.
- Compute the naive memory footprint of the three tables.
- Propose a concrete plan (hashing bucket counts, dimension choices, sharding) to bring the footprint under 20 GB while limiting damage to accuracy. Show the arithmetic.
- Why is a parameter-server style architecture (or row-wise sharding across GPUs) more appropriate here than plain data-parallel all-reduce?
Share this question