arXiv:2506.16255astro-ph.IMcs.AI2025-06中稿 · AAS Astronomical J…被引 2

用扩散模型生成真实感星系图像,兼顾外观与物理特性。

Category-based Galaxy Image Generation via Diffusion Models

  • 结合星系图像和物理属性设计新网络结构
  • 生成星系在颜色和大小分布上更一致,视觉真实
  • 支持分类生成,节省训练成本,适合天文学研究

传统星系生成依赖半解析模型和流体动力学模拟,高度依赖物理假设和参数调优。数据驱动的生成模型无需预设物理参数,可从观测数据中高效学习。其中,扩散模型在质量与多样性上优于变分自编码器(VAEs)和生成对抗网络(GANs)。本工作提出GalCatDiff,首个将星系图像特征与天体物理属性融入扩散模型网络设计的框架。其采用增强型U-Net和新型Astro-RAB(残差注意力块),动态融合注意力与卷积操作,确保全局一致性与局部细节保真。同时使用类别嵌入实现特定类型星系生成,避免为每类训练独立模型带来的高计算开销。实验表明,GalCatDiff在样本颜色与尺寸分布一致性上显著优于现有方法,生成星系兼具视觉真实性和物理一致性。该框架可提升星系模拟可靠性,并有望作为数据增强工具支持未来星系分类算法开发。

原文摘要 · Abstract (English)

Conventional galaxy generation methods rely on semi-analytical models and hydrodynamic simulations, which are highly dependent on physical assumptions and parameter tuning. In contrast, data-driven generative models do not have explicit physical parameters pre-determined, and instead learn them efficiently from observational data, making them alternative solutions to galaxy generation. Among these, diffusion models outperform Variational Autoencoders (VAEs) and Generative Adversarial Networks (GANs) in quality and diversity. Leveraging physical prior knowledge to these models can further enhance their capabilities. In this work, we present GalCatDiff, the first framework in astronomy to leverage both galaxy image features and astrophysical properties in the network design of diffusion models. GalCatDiff incorporates an enhanced U-Net and a novel block entitled Astro-RAB (Residual Attention Block), which dynamically combines attention mechanisms with convolution operations to ensure global consistency and local feature fidelity. Moreover, GalCatDiff uses category embeddings for class-specific galaxy generation, avoiding the high computational costs of training separate models for each category. Our experimental results demonstrate that GalCatDiff significantly outperforms existing methods in terms of the consistency of sample color and size distributions, and the generated galaxies are both visually realistic and physically consistent. This framework will enhance the reliability of galaxy simulations and can potentially serve as a data augmentor to support future galaxy classification algorithm development.

星系生成扩散模型天体物理图像生成

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。