Additive angular margin loss for federated learning in image classification

Abstract (excerpt). Federated learning (FL) is a method that leverages data from multiple sources to enhance deep learning models while safeguarding data privacy. This approach finds applications in diverse domains such as healthcare, entertainment, and user experience enhancement. Despite its effectiveness, FL’s performance is still low compared to centralized training methods, necessitating further improvements. Particularly in critical areas such as medicine, where accurate model predictions are crucial, conventional FL algorithms fall short compared to centralized training. This study introduces a new training approach to further improve the vanilla FL algorithm. Our proposed method takes advantage of feature-based adjustments using cosine angles by incorporating an angular margin loss function alongside the cross-entropy loss. It notably enhances the accuracy of the aggregated models on datasets such as MNIST …

Read the full paper

Recommended citation: Phuong-Nam Tran, Duc Ngoc Minh Dang, Eui-Nam Huh, Choong Seon Hong. (2025). "Additive angular margin loss for federated learning in image classification." 2025 International Conference on Information Networking (ICOIN), 274-279.
Download Paper