Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

FDRMFL: Multimodal Federated Feature Extraction Model Based on Information Maximization and Contrastive Learning

About

We propose FDRMFL, a task-driven multimodal feature extraction framework for federated regression under non-IID data distributions. Extracting predictive features from high-dimensional multimodal inputs is particularly challenging in this setting: data cannot leave each client, local samples are scarce and heterogeneously distributed, and unsupervised dimensionality reduction discards task-relevant information while federated training introduces representation drift across communication rounds. FDRMFL addresses these challenges through a unified four-term local objective: MSE prediction loss, a correlation-based mutual information surrogate that preserves dependence between the fused representation and the continuous target, a symmetric KL penalty that aligns cross-modal latent distributions before fusion, and an InfoNCE-style contrastive loss that anchors local representations to the global consensus. Experiments on three synthetic and two real-world near-infrared spectroscopy datasets under non-IID federated partitions, with comprehensive ablation and sensitivity analyses, demonstrate that each component contributes to the framework's effectiveness. FDRMFL reduces mean MSE by 33.8% relative to the best traditional baseline (PCA) and by 43.0% relative to VAE in simulation, and attains the lowest overall mean MSE among six federated algorithms including FedAvg, FedProx, MOON, SCAFFOLD, and FedBN.

Haozhe Wu• 2025

Related benchmarks

TaskDatasetResultRank
Moisture PredictionCorn
MSE0.1011
12
Oil PredictionCorn
MSE0.2578
12
Protein PredictionCorn
MSE0.2455
12
Starch PredictionCorn
MSE0.2291
12
Fat PredictionTecator Client 1
MSE0.1057
4
Fat PredictionTecator Client 2
MSE0.1444
4
Fat PredictionTecator Client 3
MSE0.1328
4
Grain composition predictionCorn Link-1 scenario (Client 1)
MSE0.9193
4
Grain composition predictionCorn Link-1 scenario (Client 2)
MSE0.9708
4
Grain composition predictionCorn Link-1 scenario (Client 3)
MSE0.8125
4
Showing 10 of 22 rows

Other info

Follow for update