Improving Fine-Tuning with Latent Cluster Correction
Machine Learning
2025-01-22 v1
Abstract
The existence of salient semantic clusters in the latent spaces of a neural network during training strongly correlates its final accuracy on classification tasks. This paper proposes a novel fine-tuning method that boosts performance by optimising the formation of these latent clusters, using the Louvain community detection algorithm and a specifically designed clustering loss function. We present preliminary results that demonstrate the viability of this process on classical neural network architectures during fine-tuning on the CIFAR-100 dataset.
Cite
@article{arxiv.2501.11919,
title = {Improving Fine-Tuning with Latent Cluster Correction},
author = {Cédric Ho Thanh},
journal= {arXiv preprint arXiv:2501.11919},
year = {2025}
}
Comments
8 pages, 4 figures, 4 tables