用无监督方法融合深度学习与概率模型,高效精准分割聚氨酯泡沫微观结构
Unsupervised Segmentation of Micro-CT Scans of Polyurethane Structures By Combining Hidden-Markov-Random Fields and a U-Net
- 结合隐马尔可夫随机场与U-Net,实现无标注数据下的快速分割
- 在μCT数据集上达到高精度,无需真实标签即可完成分割
- 提出预训练策略,显著减少有监督训练所需标注数据量
从图像中提取数字材料表征是量化材料性能的必要前提。过去虽有多种分割方法被广泛研究,但常受限于精度或速度。随着机器学习的发展,基于监督学习的卷积神经网络(CNN)在各类分割任务中表现优异,但需大量标注数据。无监督方法虽无需真实标签,却往往耗时长且精度较低。隐马尔可夫随机场(HMRF)是一种无监督分割方法,融合了邻域关系与类别分布概念。本文提出一种将HMRF理论与CNN分割相结合的新方法,兼顾无监督学习与快速分割优势。我们分析了不同邻域项与组件对无监督HMRF损失的影响。实验表明,HMRF-UNet在聚氨酯(PU)泡沫结构的微计算机断层扫描(μCT)图像数据集上实现了高精度分割,且无需真实标签。最后,我们提出并验证了一种预训练策略,显著降低了后续有监督训练所需的标注数据量。
原文摘要 · Abstract (English)
Extracting digital material representations from images is a necessary prerequisite for a quantitative analysis of material properties. Different segmentation approaches have been extensively studied in the past to achieve this task, but were often lacking accuracy or speed. With the advent of machine learning, supervised convolutional neural networks (CNNs) have achieved state-of-the-art performance for different segmentation tasks. However, these models are often trained in a supervised manner, which requires large labeled datasets. Unsupervised approaches do not require ground-truth data for learning, but suffer from long segmentation times and often worse segmentation accuracy. Hidden Markov Random Fields (HMRF) are an unsupervised segmentation approach that incorporates concepts of neighborhood and class distributions. We present a method that integrates HMRF theory and CNN segmentation, leveraging the advantages of both areas: unsupervised learning and fast segmentation times. We investigate the contribution of different neighborhood terms and components for the unsupervised HMRF loss. We demonstrate that the HMRF-UNet enables high segmentation accuracy without ground truth on a Micro-Computed Tomography ($μ$CT) image dataset of Polyurethane (PU) foam structures. Finally, we propose and demonstrate a pre-training strategy that considerably reduces the required amount of ground-truth data when training a segmentation model.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。