← Search

Altan Haan

2 accepted papers

2024

SlimFit: Memory-Efficient Fine-Tuning of Transformer-based Models Using Training Dynamics

NAACL 2024long

Transformer-based models, such as BERT and ViT, have achieved state-of-the-art results across different natural language processing (NLP) and computer vision (CV) tasks. However, these models are extremely memory intensive during their fine-tuning process, making them difficult to deploy on GPUs wit…

2021

Dynamic Tensor Rematerialization

ICLR 2021spotlight

Checkpointing enables the training of deep learning models under restricted memory budgets by freeing intermediate activations from memory and recomputing them on demand. Current checkpointing techniques statically plan these recomputations offline and assume static computation graphs. We demonstrat…