Hyrax框架助天文学家快速探索海量天文数据,无需标注也能发现新天体。
Hyrax: An Extensible Framework for Rapid ML Experimentation and Unsupervised Discovery in the Era of Rubin, Roman, and Euclid

- 模块化设计支持数据到推理全流程,集成向量库与三维可视化
- 在40万星系中无监督发现合并星系和低亮度候选体,识别成像伪影
- 适合处理下一代巡天数据的科研人员,尤其擅长无监督探索
美国国家科学基金会-能源部维拉·C·鲁宾天文台、罗曼空间望远镜、欧几里得任务及其他新一代巡天项目将产生大规模成像、光谱和时域数据,使天文机器学习瓶颈从模型设计转向基础设施。我们提出Hyrax,一个开源、模块化、支持GPU的Python框架,涵盖天文学全链条机器学习流程:从数据获取、训练到推理与实验对比。其功能包括多模态数据支持、集成向量数据库实现相似性搜索,以及交互式二维/三维潜在空间探索,用于无监督发现。我们在真实巡天数据上展示五个代表性应用:(i) 在约4×10⁵个鲁宾LSST数据预览1(DP1)星系上进行无监督表征学习,发现了未被欧几里得和暗能量巡天目录收录的合并星系与低表面亮度候选体,同时分离出成像伪影;(ii) 混合密度聚类识别DP1数据中的大尺度引力透镜候选体;(iii) 利用齐基瞬变源巡天的光变曲线、光谱、图像和元数据进行多模态早期暂现源分类;(iv) 在暗能量相机黄道探测项目中对移位叠加搜寻的假阳性进行监督过滤;(v) 通过合成源注入,在超广角相机和类LSST图像中监督检测半解析矮星系。这些结果表明,Hyrax为下一代天文巡天提供了专用机器学习基础设施,支持系统性发现与快速方法迭代。
原文摘要 · Abstract (English)
The NSF-DOE Vera C. Rubin Observatory, Roman Space Telescope, Euclid, and other next-generation surveys will deliver imaging, spectroscopic, and time-domain data at scales that increasingly shift the bottleneck in astronomical machine learning (ML) projects from model design to infrastructure. We present Hyrax, an open-source, modular, GPU-enabled Python framework that supports the full ML lifecycle in astronomy: from data acquisition and training to inference and experiment comparison, with capabilities including multimodal dataset support, integrated vector databases for similarity search, and interactive two- and three-dimensional latent-space exploration for unsupervised discovery. We demonstrate Hyrax's versatility through five representative applications on real survey data: (i) unsupervised representation learning on $\sim 4\times10^5$ Rubin Legacy Survey of Space and Time (LSST) Data Preview 1 (DP1) galaxies, surfacing new merger and low-surface-brightness candidates missing from reference Euclid and Dark Energy Survey catalogs, while also isolating imaging artifacts -- all without labeled training data; (ii) hybrid density-based clustering for identifying cluster-scale gravitational lens candidates in DP1 data; (iii) multimodal early-time transient classification in the Zwicky Transient Facility leveraging light curves, spectra, images, and metadata; (iv) supervised false-positive filtering in shift-and-stack searches for distant solar system objects in the Dark Energy Camera Ecliptic Exploration Project survey; and (v) supervised detection of semi-resolved dwarf galaxies in Hyper Suprime-Cam and LSST-like imaging using synthetic source injection. Together, these results demonstrate that Hyrax provides astronomy-specific ML infrastructure that enables systematic discovery and rapid methodological iteration across next-generation astronomical surveys.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。