Resubmission in progress · 2026
A Unified ML Pipeline for Lunar Terrain Age Determination
An end-to-end ML pipeline that classifies and clusters 1.3M+ lunar impact craters as a scalable alternative to manual crater-counting.
First author
Core idea
Lunar terrain age is traditionally estimated by manually counting impact craters — a slow, labor-intensive process. This work builds a scalable, automated alternative: a two-stage machine learning pipeline that first identifies genuine primary craters from a large, noisy catalog, then groups them into age-based clusters.
Methodology
- Classification — a supervised model separates true primary impact craters from secondary craters and catalog noise across a dataset of 1.3M+ candidate craters.
- Clustering — the confirmed primary craters are grouped into age-based clusters using an ensemble of unsupervised methods, combined by majority voting for robustness against any single algorithm's assumptions.
- Validation — results are checked against known, named lunar terrains with independently established ages, including Mare Imbrium and Mare Frigoris.
Architecture
- Classifier: a ResNet-18-based model, achieving 95.6% validation accuracy at identifying primary craters, isolating roughly 1.06M primary craters from the original 1.3M+ candidates.
- Clustering ensemble: k-means, DBSCAN, HDBSCAN, and Gaussian Mixture Models, combined via majority voting into 4 age-based groups.
Results
The pipeline's age groupings align with the known relative ages of validated lunar terrains, supporting it as a scalable, reproducible alternative to manual crater-counting for large-scale lunar surface analysis.
Status
This paper was accepted at ICAITA 2026 (Beijing), but I wasn't able to attend in person to present it, so it won't be published from that track. I'm currently identifying a new venue to resubmit it to.