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.

Ссылки и действия