Data curation via joint example selection further accelerates multimodal learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Evans, Talfan, Parthasarathy, Nikhil, Merzic, Hamza, Henaff, Olivier J.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917704741421056
author Evans, Talfan
Parthasarathy, Nikhil
Merzic, Hamza
Henaff, Olivier J.
author_facet Evans, Talfan
Parthasarathy, Nikhil
Merzic, Hamza
Henaff, Olivier J.
contents Data curation is an essential component of large-scale pretraining. In this work, we demonstrate that jointly selecting batches of data is more effective for learning than selecting examples independently. Multimodal contrastive objectives expose the dependencies between data and thus naturally yield criteria for measuring the joint learnability of a batch. We derive a simple and tractable algorithm for selecting such batches, which significantly accelerate training beyond individually-prioritized data points. As performance improves by selecting from larger super-batches, we also leverage recent advances in model approximation to reduce the associated computational overhead. As a result, our approach--multimodal contrastive learning with joint example selection (JEST)--surpasses state-of-the-art models with up to 13$\times$ fewer iterations and 10$\times$ less computation. Essential to the performance of JEST is the ability to steer the data selection process towards the distribution of smaller, well-curated datasets via pretrained reference models, exposing the level of data curation as a new dimension for neural scaling laws.
format Preprint
id arxiv_https___arxiv_org_abs_2406_17711
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Data curation via joint example selection further accelerates multimodal learning
Evans, Talfan
Parthasarathy, Nikhil
Merzic, Hamza
Henaff, Olivier J.
Machine Learning
Artificial Intelligence
Data curation is an essential component of large-scale pretraining. In this work, we demonstrate that jointly selecting batches of data is more effective for learning than selecting examples independently. Multimodal contrastive objectives expose the dependencies between data and thus naturally yield criteria for measuring the joint learnability of a batch. We derive a simple and tractable algorithm for selecting such batches, which significantly accelerate training beyond individually-prioritized data points. As performance improves by selecting from larger super-batches, we also leverage recent advances in model approximation to reduce the associated computational overhead. As a result, our approach--multimodal contrastive learning with joint example selection (JEST)--surpasses state-of-the-art models with up to 13$\times$ fewer iterations and 10$\times$ less computation. Essential to the performance of JEST is the ability to steer the data selection process towards the distribution of smaller, well-curated datasets via pretrained reference models, exposing the level of data curation as a new dimension for neural scaling laws.
title Data curation via joint example selection further accelerates multimodal learning
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2406.17711