Toward Class Imbalance and Uncertainty in Powder XRD Analysis: A Dual-Channel Fusion Network for Space Group Classification.
Accurate identification of space groups from powder X-ray diffraction (pXRD) is essential for understanding crystal structures and accelerating materials discovery. However, this task remains highly challenging due to inherent peak overlap, experimental noise, and the complexity of the 230-class classification problem. To address the critical issues of class imbalance and data scarcity, we first design a general physics-informed data augmentation pipeline. We then propose a dual-channel fusion uncertainty-aware network (DFUN) for automated space group classification. The DFUN architecture integrates two complementary feature representations: convolutional features extracted directly from raw diffraction profiles and domain-specific peak descriptors. These distinct representations are adaptively fused through a gating mechanism. Furthermore, to mitigate the inherent long-tailed distribution of crystallographic data, we employ a hybrid loss function that combines Focal Loss with Label Smoothing. Finally, we incorporate Monte Carlo Dropout to provide predictive uncertainty estimation, thereby enabling not only accurate classification but also a crucial assessment of the model's reliability. Evaluated on large-scale simulated data and two public data sets (opXRD and RRUFF), DFUN outperforms the evaluated baseline methods across the reported metrics. The framework also provides uncertainty-aware predictions, establishing DFUN as a robust and interpretable solution for high-throughput automated crystallographic analysis from powder diffraction.