Intelligent Detection of Lunar Impact Craters Using DEM and Gravity Data Based on ResNet and Vision Transformer

The craters on the moon hold important clues about the history of impacts in our solar system. To address the limitation of traditional intelligent methods in detecting buried craters, this study proposes a novel intelligent detection approach based on DEM and gravity data. We designed a hybrid network architecture (ResNet + ViT) that combines the local feature extraction strengths of Convolutional Neural Networks with the global context modeling capabilities of Vision Transformer. By combining the complementary information from DEM and gravity anomaly data, it achieves comprehensive detection of lunar craters—from those visible on the surface to buried subsurface structures. To mitigate the inherent sample imbalance in both gravity anomaly and DEM training data, we employ a U-Net architecture augmented with residual blocks and train it using a Focal Loss function with dynamic focusing parameters. Experimental results show that: (1) The proposed method attains high segmentation accuracy, achieving a mean Intersection over Union of 81.3% on the DEM test set and 82.6% on the gravity anomaly test set, respectively. (2) Our method outperforms U-Net and its mainstream variants, achieving a precision of 89.48% and superior detection completeness. (3) Application to representative geological units, including the Wugang Basin, Archimedes Crater, and Mare Moscoviense, validates the robustness and practical utility of our method. This study, thus, provides a novel technical framework for global-scale mapping of lunar impact craters and yields new insights into the evolutionary history of the lunar surface.