Scaling data ingestion for machine learning training at Meta
Sep 19, 2022Aarti Basant

Summary
This post details the evolution of Meta's data ingestion infrastructure for machine learning training. It introduces a disaggregated Data PreProcessing tier (DPP) that handles data fetching, decryption, feature extraction, tensor conversion, and last-mile transformations. Key optimizations include feature flattening for efficient selective feature reading (2-2.3x efficiency gains), client-side rebatching to support larger batch sizes and improve reader node throughput (20-40% improvement), and the FlatMaps project to optimize memory bandwidth utilization in data reading workloads. These improvements address the mismatch between hardware performance growth and data ingestion demands, and the power constraints in data centers.
More from the AI Research and Development trail
- An Organizational Second Brain: Building an AI That Learns From Experts
- From User Sequences to Scaling Laws: A Multi-Stage Architecture for Meta’s Ads Ranking
- GEM Training: How Meta Doubled the Efficiency of Its LLM-Scale Ads Foundation Model
- Exploring Hierarchical Interest Representation For Meta Ads Deep Funnel Optimization
- Meta’s AI Storage Blueprint at Scale