GeoCFM: Positive-Only Conditional Flow Matching for Mineral Occurrence Sampling
Organizations: Faculty of Computer and Information Science, Ben-Gurion University of the Negev, Israel · Department of Earth, Ocean and Atmospheric Sciences, University of British Columbia, Canada
Abstract
Critical mineral discovery is a positive-only problem: deposits are observed as sparse locations, while unlabeled regions are not reliable negatives, and similar geophysical signatures can arise from different subsurface states. We therefore model mineral targeting as learning a conditional spatial distribution over occurrence locations, , given geo-images , rather than predicting a deterministic per-pixel score map. We introduce GeoCFM, a conditional flow-matching model that generates mineral occurrence point sets conditioned on multi-channel geo-images; GeoCFM learns a point-wise transport field in , using UNet features with point-conditioned velocity prediction to bridge dense rasters and sparse supervision without pseudo-negatives. On a synthetic magnetics--geochemistry benchmark with latent activation and on USGS Earth MRI data with a spatially disjoint tile split, GeoCFM improves geometric agreement with observed occurrences over score-map and non-conditional baselines, while representing epistemic uncertainty through conditional sampling.
Figures & tables
| Method | CD | Sink. | F@5 | NLL | Top5 |
|---|---|---|---|---|---|
| UNet-Seg | |||||
| GBDT | |||||
| RF | |||||
| Poisson-LR | |||||
| OCSVM | |||||
| Retrieval-KDE |
| Method | CD | Sink. | F@5 | NLL | Top5 |
|---|---|---|---|---|---|
| UNet-Seg | |||||
| GBDT | |||||
| RF | |||||
| Poisson-LR | |||||
| OCSVM | |||||
| Retrieval-KDE |
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
| Setting | Synthetic | Earth MRI |
|---|---|---|
| Input channels | 2 ( ) | 8 geophysical layers |
| Base UNet width | 32 | 48 |
| UNet depth | 3 | 3 |
| Feature dimension | 64 | 128 |
| Time-embedding dimension | 64 | 64 |
| Velocity head | residual network | MLP |
| Setting | Synthetic | Earth MRI |
|---|---|---|
| Batch size | 4 | 8 |
| Learning rate | ||
| Weight decay | ||
| Maximum iterations | 500,000 | 400,000 |
| Gradient clip | 1.0 | 1.0 |
| Validation interval | 250 iterations | 2,000 iterations |
| Synthetic | Earth MRI | |||
|---|---|---|---|---|
| Method | Train CD | Test CD | Train CD | Test CD |
| Uniform sampling | 49.30 | 49.61 | 63.99 | 64.25 |
| Global KDE | 37.12 | 41.16 | 57.10 | 66.35 |
| Poisson-LR | 38.75 | 39.93 | 62.80 | 64.21 |
| Random Forest | 32.08 | 37.09 | 49.12 | 54.37 |
| UNet-Seg | 26.11 | 29.46 | 41.30 | 46.81 |
| Data | Pos. frac. | CD | Sink. | NLL | F@5 |
|---|---|---|---|---|---|
| Synthetic | |||||
| Synthetic | |||||
| Synthetic | |||||
| Earth MRI | |||||
| Earth MRI | |||||
| Earth MRI |
| Data | Method | ms/patch | CD | F@5 |
|---|---|---|---|---|
| Synthetic | UNet-Seg | 13.92 | 29.46 | 0.615 |
| Synthetic | GeoCFM | 19.79 | 9.30 | 0.580 |
| Synthetic | GeoCFM | 21.98 | 9.39 | 0.610 |
| Synthetic | GeoCFM | 38.47 | 9.37 | 0.634 |
| Earth MRI | UNet-Seg | 12.50 | 46.81 | 0.085 |
| Earth MRI | GeoCFM | 16.95 | 12.00 | 0.140 |
| Budget | 100 | 250 | 500 | 1000 | 2000 |
|---|---|---|---|---|---|
| Synthetic NLL | 11.79 | 10.96 | 10.79 | 10.66 | 10.56 |
| Synthetic stab. | 0.894 | 0.946 | 0.970 | 0.982 | 0.988 |
| Earth MRI NLL | 10.20 | 9.32 | 9.24 | 9.02 | 8.99 |
| Earth MRI stab. | 0.876 | 0.962 | 0.971 | 0.984 | 0.991 |