BCDDM: Branch-Corrected Denoising Diffusion Model for Black Hole Image Generation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: liu, Ao, Zhang, Zelin, Chen, Songbai, Wen, Cuihong, Wang, Jieci
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915305862725632
author liu, Ao
Zhang, Zelin
Chen, Songbai
Wen, Cuihong
Wang, Jieci
author_facet liu, Ao
Zhang, Zelin
Chen, Songbai
Wen, Cuihong
Wang, Jieci
contents The properties of black holes and accretion flows can be inferred by fitting Event Horizon Telescope (EHT) data to simulated images generated through general relativistic ray tracing (GRRT). However, due to the computationally intensive nature of GRRT, the efficiency of generating specific radiation flux images needs to be improved. This paper introduces the Branch Correction Denoising Diffusion Model (BCDDM), a deep learning framework that synthesizes black hole images directly from physical parameters. The model incorporates a branch correction mechanism and a weighted mixed loss function to enhance accuracy and stability. We have constructed a dataset of 2,157 GRRT-simulated images for training the BCDDM, which spans seven key physical parameters of the radiatively inefficient accretion flow (RIAF) model. Our experiments show a strong correlation between the generated images and their physical parameters. By enhancing the GRRT dataset with BCDDM-generated images and using ResNet50 for parameter regression, we achieve significant improvements in parameter prediction performance. BCDDM offers a novel approach to reducing the computational costs of black hole image generation, providing a faster and more efficient pathway for dataset augmentation, parameter estimation, and model fitting.
format Preprint
id arxiv_https___arxiv_org_abs_2502_08528
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle BCDDM: Branch-Corrected Denoising Diffusion Model for Black Hole Image Generation
liu, Ao
Zhang, Zelin
Chen, Songbai
Wen, Cuihong
Wang, Jieci
Astrophysics of Galaxies
Computer Vision and Pattern Recognition
The properties of black holes and accretion flows can be inferred by fitting Event Horizon Telescope (EHT) data to simulated images generated through general relativistic ray tracing (GRRT). However, due to the computationally intensive nature of GRRT, the efficiency of generating specific radiation flux images needs to be improved. This paper introduces the Branch Correction Denoising Diffusion Model (BCDDM), a deep learning framework that synthesizes black hole images directly from physical parameters. The model incorporates a branch correction mechanism and a weighted mixed loss function to enhance accuracy and stability. We have constructed a dataset of 2,157 GRRT-simulated images for training the BCDDM, which spans seven key physical parameters of the radiatively inefficient accretion flow (RIAF) model. Our experiments show a strong correlation between the generated images and their physical parameters. By enhancing the GRRT dataset with BCDDM-generated images and using ResNet50 for parameter regression, we achieve significant improvements in parameter prediction performance. BCDDM offers a novel approach to reducing the computational costs of black hole image generation, providing a faster and more efficient pathway for dataset augmentation, parameter estimation, and model fitting.
title BCDDM: Branch-Corrected Denoising Diffusion Model for Black Hole Image Generation
topic Astrophysics of Galaxies
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2502.08528