arXiv:2508.00896cs.CVcond-mat.mtrl-sci2025-08

用相分数控制生成微结构图像和分割图,解决金属显微图像标注数据少的问题。

Phase-fraction guided denoising diffusion model for augmenting multiphase steel microstructure segmentation via micrograph image-mask pair synthesis

  • 通过相分数向量控制扩散模型,一次生成图像与对应分割掩码。
  • 在稀有相上分割准确率提升,优于传统增强和两阶段生成模型。
  • 适合金属显微图像数据稀缺场景,可快速生成高质量训练数据。

机器学习在金属显微组织分割中的效果常受限于人工标注的相掩码不足,尤其针对合金中罕见或复杂形态。本文提出PF-DiffSeg,一种基于相分数控制的一阶段去噪扩散框架,通过单一生成轨迹联合合成微结构图像及其对应分割掩码,以提升分割精度。通过条件化全局相分数向量,增强对少数类别的表示,使生成样本在成分上合理、结构上连贯,从而提高数据多样性与训练效率。在增材制造多相钢的MetalDAM基准上评估,该方法相比标准增强策略显著提升少数相的分割准确率,并优于两阶段掩码引导扩散模型与生成对抗网络(GAN)基线,同时推理时间更短。该方法将生成与条件控制整合为统一框架,为金属学应用提供可扩展的数据增强方案。

原文摘要 · Abstract (English)

The effectiveness of machine learning in metallographic microstructure segmentation is often constrained by the lack of human-annotated phase masks, particularly for rare or compositionally complex morphologies within the metal alloy. We introduce PF-DiffSeg, a phase-fraction controlled, one-stage denoising diffusion framework that jointly synthesizes microstructure images and their corresponding segmentation masks in a single generative trajectory to further improve segmentation accuracy. By conditioning on global phase-fraction vectors, augmented to represent real data distribution and emphasize minority classes, our model generates compositionally valid and structurally coherent microstructure image and mask samples that improve both data diversity and training efficiency. Evaluated on the MetalDAM benchmark for additively manufactured multiphase steel, our synthetic augmentation method yields notable improvements in segmentation accuracy compared to standard augmentation strategies especially in minority classes and further outperforms a two-stage mask-guided diffusion and generative adversarial network (GAN) baselines, while also reducing inference time compared to conventional approach. The method integrates generation and conditioning into a unified framework, offering a scalable solution for data augmentation in metallographic applications.

金属显微扩散模型数据增强分割

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