Powered by OpenAIRE graph
Found an issue? Give us feedback
image/svg+xml art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos Open Access logo, converted into svg, designed by PLoS. This version with transparent background. http://commons.wikimedia.org/wiki/File:Open_Access_logo_PLoS_white.svg art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos http://www.plos.org/ ZENODOarrow_drop_down
image/svg+xml art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos Open Access logo, converted into svg, designed by PLoS. This version with transparent background. http://commons.wikimedia.org/wiki/File:Open_Access_logo_PLoS_white.svg art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos http://www.plos.org/
ZENODO
Doctoral thesis . 2026
License: CC BY
Data sources: ZENODO
ZENODO
Thesis . 2026
License: CC BY
Data sources: Datacite
ZENODO
Thesis . 2026
License: CC BY
Data sources: Datacite
versions View all 2 versions
addClaim

VRAM OPTIMIZATION FOR TRANSFORMERS BASED NUTRITION PREDICTION MODEL

Authors: Duong, Tien;

VRAM OPTIMIZATION FOR TRANSFORMERS BASED NUTRITION PREDICTION MODEL

Abstract

Accurate estimation of nutritional content from food images can promote better dietary awareness and health management. Recent advances in deep learning have enabled transformer-based models to achieve state-of-the-art results for nutrition prediction; however, their high video memory (VRAM) requirements limit accessibility and scalability. In this work, we propose the integration of four VRAM optimization techniques into an existing state-of-the-art transformer-based nutrition prediction framework to reduce memory consumption during training. The optimization methods include Automatic Mixed Precision (AMP), gradient checkpointing, gradient accumulation, and the Adam8bit optimizer. The framework employs a Swin Transformer (Swin-T) backbone with a Feature Pyramid Network (FPN) and a Swin-TUNA segmentation module for food-region detection. Experiments on the Nutrition5k dataset demonstrate a 77% reduction in peak VRAM and a 78% reduction in reserved VRAM during training, while maintaining model accuracy (PMAE = 16.9%, improved from 17.2%). Although training time increased due to re-computation and accumulation overhead, the proposed optimization approach enables stable training of large transformer architectures on mid-range GPUs. This work provides a practical direction for developing memory-efficient nutrition prediction systems, improving the accessibility of transformer-based models for future research and deployment.

Keywords

VRAM optimization, nutrient estimation, deep learning, dietary assessment, transformers, memory-efficient training

  • BIP!
    Impact byBIP!
    selected citations
    These citations are derived from selected sources.
    This is an alternative to the "Influence" indicator, which also reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
    0
    popularity
    This indicator reflects the "current" impact/attention (the "hype") of an article in the research community at large, based on the underlying citation network.
    Average
    influence
    This indicator reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
    Average
    impulse
    This indicator reflects the initial momentum of an article directly after its publication, based on the underlying citation network.
    Average
Powered by OpenAIRE graph
Found an issue? Give us feedback
selected citations
These citations are derived from selected sources.
This is an alternative to the "Influence" indicator, which also reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
BIP!Citations provided by BIP!
popularity
This indicator reflects the "current" impact/attention (the "hype") of an article in the research community at large, based on the underlying citation network.
BIP!Popularity provided by BIP!
influence
This indicator reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
BIP!Influence provided by BIP!
impulse
This indicator reflects the initial momentum of an article directly after its publication, based on the underlying citation network.
BIP!Impulse provided by BIP!
0
Average
Average
Average
Green