about
Tf.data: A Machine Learning Data Processing Framework (arxiv.org)
14 points by Anon84 on Jan 31, 2021 | hide | past | pdf | discuss on HN

In plain words: A system for building the pipelines that feed data to training: users combine reusable pieces while the runtime keeps them fast. It kept training fast without hand-tuning, and a study of millions of Google jobs found data loading eats a big share of their resources.

Abstract · tf.data: A Machine Learning Data Processing Framework

Training machine learning models requires feeding input data for models to ingest. Input pipelines for machine learning jobs are often challenging to implement efficiently as they require reading large volumes of data, applying complex transformations, and transferring data to hardware accelerators while overlapping computation and communication to achieve optimal performance. We present tf.data, a framework for building and executing efficient input pipelines for machine learning jobs. The tf.data API provides operators which can be parameterized with user-defined computation, composed, and reused across different machine learning domains. These abstractions allow users to focus on the application logic of data processing, while tf.data's runtime ensures that pipelines run efficiently. We demonstrate that input pipeline performance is critical to the end-to-end training time of state-of-the-art machine learning models. tf.data delivers the high performance required, while avoiding the need for manual tuning of performance knobs. We show that tf.data features, such as parallelism, caching, static optimizations, and non-deterministic execution are essential for high performance. Finally, we characterize machine learning input pipelines for millions of jobs that ran in Google's fleet, showing that input data processing is highly diverse and consumes a significant fraction of job resources. Our analysis motivates future research directions, such as sharing computation across jobs and pushing data projection to the storage layer.

Derek G. Murray, Jiri Simsa, Ana Klimovic, Ihor Indyk
arXiv:2101.12127 · cs.LG, cs.MS · submitted Jan 28, 2021 · updated Feb 23, 2021
abstract · pdf · html

add comment on HN