Asteria 🌌

Asteria is a lightweight, high-performance deep learning inference engine and instance deblender for stars in astronomical imaging, powered by boundary-aware UNet++ (Nested U-Net).

Designed for the Nancy Grace Roman Space Telescope, Asteria provides promptless, end-to-end star detection in a single neural forward pass.

_{Example of Asteria star detection on a simulated Roman WFI exposure.}

Key Features

• Lightweight & Self-Contained: Pure PyTorch UNet++ architecture without bulky dependencies or heavyweight vision foundation models.
• Hardware Acceleration with Automatic Fallback: Auto-detects and optimizes execution across NVIDIA CUDA (with TF32 and AMP), Apple Silicon MPS (Metal Performance Shaders on macOS), and universal CPU.
• Physical Boundary-Aware Deblending: Trained with 3-class supervision:
  • 0: Sky Background
  • 1: Source Interior / Core
  • 2: Deblending Boundary / Contact Interface
• Seamless Tiled Dense Inference (predict_dense): Sliding-window 2D Hann/cosine window blending to process arbitrary image sizes (from local cutouts up to full $4088 \times 4088$ Roman WFI exposures) with bounded memory ($< 350$ MB VRAM).
• Photometric LeakySB Stretch: Non-linear surface brightness transformation that preserves stellar point-spread functions (PSFs) while lifting faint, diffuse galactic wings.
• Unified Astronomical I/O & Cloud Storage: Direct support for 2D astronomical FITS (.fits, .fits.gz), Roman ASDF (.asdf Level 2 calibrated exposures and Level 3 mosaics via roman_datamodels), raw NumPy arrays, and Amazon Web Services (AWS) S3 URLs (s3://bucket/key... or HTTPS S3 URLs). Automatically converts Level 2 exposures to surface brightness units ($\text{MJy/sr}$) using calibrated photometry and enables $\text{ZP}=20.472$.

Installation

From PyPI

pip install asteria-roman

From Source

git clone https://github.com/PabloMSanAla/asteria.git
cd asteria
pip install -e .

Quickstart

1. Python API

import asteria

# 1. Load pre-trained model (auto-downloads weights on first run, auto-selects device)
segmentor = asteria.load_model("latest")

# 2. Run dense 3-class segmentation inference on a local file or AWS S3 URL
pred_mask, probs, stretched = segmentor.predict_dense(
    "s3://my-roman-bucket/exposures/r0000101_wfi.asdf",  # or local "scene_0001_F106.fits"
    tile_size=512,
    overlap=64,
)

# 3. Render publication-ready side-by-side 2-panel figure
fig = asteria.plot_two_panel(
    stretched=stretched,
    pred_mask=pred_mask,
    save_path="prediction.png",
)

2. Command-Line Interface (CLI)

Asteria includes the asteria (and asteria-predict) CLI command:

# Pre-download official weights for offline/cloud deployment
asteria --download

# Basic prediction and display (local file or S3 URI)
asteria s3://my-roman-bucket/exposures/r0000101_wfi.asdf

# Full-field tiled inference with saving and FITS export
asteria scene.fits \
    --tile_size 512 \
    --overlap 64 \
    --output results.png \
    --export_fits segmap.fits \
    --no_display

Model Architecture & Weights History

For comprehensive architectural details and benchmark comparisons, see the Asteria Documentation.

License

This project is licensed under the MIT License - see the LICENSE file for details.