English

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.

Keywords

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

R2 v1 2026-06-28T21:12:07.041Z