ICML2026
SymSpectra: Symmetric Information Bottleneck Framework for Molecular Structure Recognition under Imbalanced Settings
Xiaohan Qin, Wenjie Du, Yang Wang
摘要
Identifying molecular structures from spectral data is essential for early-stage chemical analysis, yet it remains a difficult task due to severe functional group imbalance and complex inter-group dependencies, which often cause existing methods to overfit frequent groups while underperforming on rare ones. To address these issues, we present SymSpectra , a Sym metric Conditional Information Bottleneck (SCIB) framework designed to seamlessly integrate multi-modal Spectra features. Our model employs the SCIB framework to fuse multi-modal spectroscopic data into a unified representation, effectively preserving discriminative signals while mitigating redundancy. To enhance robustness against data imbalance, we incorporate conditional mutual information into the training objective, increasing the model’s sensitivity to rare functional groups and challenging molecular cases. Additionally, a specialized module captures the dependencies among functional groups, improving both prediction accuracy and chemically meaningful interpretability. Experiments on multimodal spectral datasets show that SymSpectra outperforms state-of-the-art methods, achieving an F1-score of 0.970 in substructure classification and demonstrating strong robustness under various imbalance settings.