Work

Improving Fine-Tuning with Latent Cluster Correction

Year: 2025
Type: preprint
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.
Institution: RIKEN
Language: English
Open Access status: green
OpenAlex ID: W4406745230
Cites:
Cited by:
Related to: 10