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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Moisture Prediction | Corn | MSE0.1011 | 12 | |
| Oil Prediction | Corn | MSE0.2578 | 12 | |
| Protein Prediction | Corn | MSE0.2455 | 12 | |
| Starch Prediction | Corn | MSE0.2291 | 12 | |
| Fat Prediction | Tecator Client 1 | MSE0.1057 | 4 | |
| Fat Prediction | Tecator Client 2 | MSE0.1444 | 4 | |
| Fat Prediction | Tecator Client 3 | MSE0.1328 | 4 | |
| Grain composition prediction | Corn Link-1 scenario (Client 1) | MSE0.9193 | 4 | |
| Grain composition prediction | Corn Link-1 scenario (Client 2) | MSE0.9708 | 4 | |
| Grain composition prediction | Corn Link-1 scenario (Client 3) | MSE0.8125 | 4 |