LEMoN: Label Error Detection using Multimodal Neighbors

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhang, Haoran, Balagopalan, Aparna, Oufattole, Nassim, Jeong, Hyewon, Wu, Yan, Zhu, Jiacheng, Ghassemi, Marzyeh
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866912413406724096
author Zhang, Haoran
Balagopalan, Aparna
Oufattole, Nassim
Jeong, Hyewon
Wu, Yan
Zhu, Jiacheng
Ghassemi, Marzyeh
author_facet Zhang, Haoran
Balagopalan, Aparna
Oufattole, Nassim
Jeong, Hyewon
Wu, Yan
Zhu, Jiacheng
Ghassemi, Marzyeh
contents Large repositories of image-caption pairs are essential for the development of vision-language models. However, these datasets are often extracted from noisy data scraped from the web, and contain many mislabeled instances. In order to improve the reliability of downstream models, it is important to identify and filter images with incorrect captions. However, beyond filtering based on image-caption embedding similarity, no prior works have proposed other methods to filter noisy multimodal data, or concretely assessed the impact of noisy captioning data on downstream training. In this work, we propose, theoretically justify, and empirically validate LEMoN, a method to identify label errors in image-caption datasets. Our method leverages the multimodal neighborhood of image-caption pairs in the latent space of contrastively pretrained multimodal models to automatically identify label errors. Through empirical evaluations across eight datasets and twelve baselines, we find that LEMoN outperforms the baselines by over 3% in label error detection, and that training on datasets filtered using our method improves downstream captioning performance by more than 2 BLEU points over noisy training.
format Preprint
id arxiv_https___arxiv_org_abs_2407_18941
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle LEMoN: Label Error Detection using Multimodal Neighbors
Zhang, Haoran
Balagopalan, Aparna
Oufattole, Nassim
Jeong, Hyewon
Wu, Yan
Zhu, Jiacheng
Ghassemi, Marzyeh
Computer Vision and Pattern Recognition
Machine Learning
Large repositories of image-caption pairs are essential for the development of vision-language models. However, these datasets are often extracted from noisy data scraped from the web, and contain many mislabeled instances. In order to improve the reliability of downstream models, it is important to identify and filter images with incorrect captions. However, beyond filtering based on image-caption embedding similarity, no prior works have proposed other methods to filter noisy multimodal data, or concretely assessed the impact of noisy captioning data on downstream training. In this work, we propose, theoretically justify, and empirically validate LEMoN, a method to identify label errors in image-caption datasets. Our method leverages the multimodal neighborhood of image-caption pairs in the latent space of contrastively pretrained multimodal models to automatically identify label errors. Through empirical evaluations across eight datasets and twelve baselines, we find that LEMoN outperforms the baselines by over 3% in label error detection, and that training on datasets filtered using our method improves downstream captioning performance by more than 2 BLEU points over noisy training.
title LEMoN: Label Error Detection using Multimodal Neighbors
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2407.18941