Investigating the Impact of Large-Scale Pre-training on Nutritional Content Estimation from 2D Images
2508.03996v1
cs.CV
2025-08-09
Авторы:
Michele Andrade, Guilherme A. L. Silva, Valéria Santos, Gladston Moreira, Eduardo Luz
Резюме на русском
Оценка состава пищи по изображениям 2D представляет собой важную задачу с значительными последствиями для мониторинга диет и здоровья. Тем не менее, ее выполнение сложно из-за разнообразия в формате представления пищи, изменений освещения и трудности оценки объема и массы без глубины. Доступность данных для предварительного обучения (pre-training) также ограничивает реплицируемость современных методов. В данной работе мы исследуем влияние объема и свойств данных для pre-training на модели Vision Transformer (ViT), применяемые к задаче оценки состава пищи из 2D-изображений. Мы обучаем и экспериментируем с моделями ViT, pre-trained на ImageNet, COYO и JFT-300M, сравнивая их с CNN-моделями и современным методом на JFT-300M. На Nutrition5k датасете показатели Mean Absolute Error (MAE) и Mean Absolute Percentage Error (MAE%) показали, что модели pre-trained на JFT-300M оказались эффективнее. Интересно, что COYO, несмотря на большой размер, дал худшие результаты в контексте этой задачи. Наши результаты подтверждают важность качества и доменной релевантности данных для pre-training в задачах подобного рода.
Abstract
Estimating the nutritional content of food from images is a critical task
with significant implications for health and dietary monitoring. This is
challenging, especially when relying solely on 2D images, due to the
variability in food presentation, lighting, and the inherent difficulty in
inferring volume and mass without depth information. Furthermore,
reproducibility in this domain is hampered by the reliance of state-of-the-art
methods on proprietary datasets for large-scale pre-training. In this paper, we
investigate the impact of large-scale pre-training datasets on the performance
of deep learning models for nutritional estimation using only 2D images. We
fine-tune and evaluate Vision Transformer (ViT) models pre-trained on two large
public datasets, ImageNet and COYO, comparing their performance against
baseline CNN models (InceptionV2 and ResNet-50) and a state-of-the-art method
pre-trained on the proprietary JFT-300M dataset. We conduct extensive
experiments on the Nutrition5k dataset, a large-scale collection of real-world
food plates with high-precision nutritional annotations. Our evaluation using
Mean Absolute Error (MAE) and Mean Absolute Percentage Error (MAE%) reveals
that models pre-trained on JFT-300M significantly outperform those pre-trained
on public datasets. Unexpectedly, the model pre-trained on the massive COYO
dataset performs worse than the model pre-trained on ImageNet for this specific
regression task, refuting our initial hypothesis. Our analysis provides
quantitative evidence highlighting the critical role of pre-training dataset
characteristics, including scale, domain relevance, and curation quality, for
effective transfer learning in 2D nutritional estimation.
Ссылки и действия
Дополнительные ресурсы: