Improving Muon-Scattering Material Identification via Coarse Momentum Encoding and Unsupervised Domain Adaptation

Abstract

Cosmic-ray muon scattering has shown considerable potential for detecting nuclear materials and other dense contraband, but practical deployment remains challenging. A major difficulty arises from the coupling between material properties and muon momentum, since the broad natural momentum distribution influences the scattering angle and prevents unambiguous material identification. In this work, we propose a Coarse Momentum-Aware Domain Adaptation (CMADA) method to enable precise identification of materials. Instead of relying on high-precision momentum measurements, the proposed framework adopts coarse momentum binning combined with unsupervised domain adaptation to learn transferable scattering representations. In addition, a precision review mode based on averaging repeated samplings was proposed to further enhances identification performance. The coarse momentum binning strategy improves same-domain identification accuracy from 62.15% without momentum information to 89.52% with 5-bin momentum information, and further to 93.37% (precision review mode). Furthermore, the proposed unsupervised domain adaptation framework improves the cross-domain identification accuracy from 71.71% for the source-only baseline to 89.00% without requiring target domain labels.