Substructure Detection in Realistic Strong Lensing Systems with Machine Learning

The Big Picture

Imagine you’re trying to find a ghost by looking for the shadows it casts. You can’t see the ghost directly, but you can see how it warps the light around it, bending and distorting the shapes of things behind it. That’s what physicists do when they hunt for dark matter using gravitational lensing. A team from Harvard has now taught a neural network to spot those shadows automatically.

Dark matter makes up about 85% of all matter in the universe, yet we’ve never directly detected a single particle of it. What we have detected are its gravitational fingerprints: the way it bends light from distant galaxies, stretching them into arcs and rings when a massive galaxy sits between us and them. These dramatic distortions are called strong lensing systems, and they act as cosmic laboratories. Inside them, tiny clumps of dark matter called subhalos leave almost imperceptible distortions in the lensed images. Almost imperceptible, but not quite.

The problem is scale. We currently know of roughly a hundred such lensing systems. By 2030, telescopes like the Vera Rubin Observatory and Euclid will push that number to tens or hundreds of thousands. Traditional analysis of a single lens takes days to weeks. Researchers Arthur Tsang, Atınç Çağan Şengül, and Cora Dvorkin set out to fix this with a machine learning system that scans enormous catalogs and flags the systems most likely to harbor detectable dark matter substructure, like a triage nurse sorting patients before the doctors arrive.

Key Insight: A neural network trained on realistic simulated lenses can identify 71% of systems containing dark matter subhalos, at just a 10% false positive rate, in a fraction of the time traditional methods require.

How It Works

The core tool is a UNet, a neural network architecture originally designed for biomedical image segmentation. In its medical application, a UNet identifies tumor boundaries in an MRI scan. Here, it learns to pick out regions in a lensing image that have been nudged and distorted by a dark matter subhalo. The network takes a lensed galaxy image as input and outputs a heat map (a segmentation mask highlighting where a subhalo is most likely lurking).

Training required building a realistic simulated dataset, and this is where the work gets meticulous. The team used lenstronomy to generate thousands of synthetic lensing systems, deliberately avoiding the overly tidy simulations that undermined earlier machine learning work. Their simulations included:

• Realistic source galaxies from the COSMOS survey (real Hubble photographs, not smooth idealized shapes)
• Complex main lenses modeled with power-law elliptical potentials, external shear, and higher-order multipoles, capturing the lumpy, asymmetric mass distributions of real galaxies
• Realistic noise consistent with space telescope observations
• Subhalos modeled as truncated NFW profiles spanning 10⁸ to 10¹⁰ solar masses, placed along Einstein rings where their effect is strongest

The realism matters because earlier machine learning approaches trained on simplified lenses fell apart on real data. Using actual galaxy morphologies, realistic noise, and complex lens models goes a long way toward fixing that.

The network was trained both to classify whether an image contains a subhalo and to localize it. At a false positive rate (FPR) of 10% (meaning one in ten subhalo-free systems would trigger a false alarm), the UNet achieved a true positive rate (TPR) of 71% for subhalos in the 10⁹ to 10⁹·⁵ solar mass range. Performance improved with higher image resolution and with higher subhalo concentration, which measures how centrally dense the subhalo’s mass profile is.

The concentration result carries a warning. Real-world subhalo detections have tended to favor high concentrations, which the algorithm handles well. But standard ΛCDM (Lambda Cold Dark Matter) simulations predict that most subhalos should have lower concentrations. The algorithm struggles with these, raising an open question: are real subhalos genuinely more concentrated than simulations predict, or are we only finding the easy cases?

Why It Matters

Two problems collide here. The first is practical: the coming data flood from next-generation surveys will be impossible to analyze with traditional statistical sampling methods, which need days of computation per system. A trained neural network evaluates an entire catalog in seconds, acting as a first-pass filter that directs expensive follow-up analysis toward the most promising targets. That speed advantage isn’t incremental; it’s what makes the science possible at all.

The second problem is fundamental. Dark matter models like warm dark matter (WDM), in which particles move fast enough to smooth out small-scale clumping, and self-interacting dark matter (SIDM), in which dark matter particles collide with each other and reshape how clumps form, predict different subhalo populations than standard ΛCDM. The differences are sharpest at the smallest scales. Measuring subhalo abundance and properties across thousands of lensing systems would test these competing theories directly.

The concentration sensitivity problem identified here is itself a clue. Either our simulations are missing something about how subhalos form and evolve, or detections to date are a biased sample, the equivalent of concluding all fish are large because your net only catches large ones. Resolving this will sharpen our picture of what dark matter actually is.

Bottom Line: By applying a UNet to realistically simulated strong lenses, Tsang et al. show that machine learning can efficiently screen thousands of lensing systems for dark matter substructure. That capability will be essential when next-generation surveys deliver their enormous catalogs, and it could reshape our understanding of dark matter’s small-scale behavior.