arXiv:2606.21061cs.CV2026-06

通过采样不同网络结构实现图像分割多样性,提升医疗影像等关键场景的可靠性。

Neural Architecture Distributions: A New Paradigm for Stochastic Segmentation

论文配图:Neural Architecture Distributions: A New Paradigm for Stochastic Segmentation
图 1 · 摘自论文原文
  • 用可搜索位置的离散架构分布代替传统噪声或隐变量,实现随机性
  • 在LIDC-IDRI数据集上达到最优分布匹配和假设覆盖度,优于现有方法
  • 支持架构溯源,适合需可解释性的医学与工业质检场景

随机分割旨在为单张图像生成多个合理掩码,对医疗影像或建筑缺陷检测等高安全、高精度场景至关重要。现有方法多通过注入连续隐变量或迭代去噪轨迹引入随机性,其来源难以直接搜索或审计。本文提出以架构分布作为新的随机源:不在隐空间采样,而是从分割主干中多个可搜索位置的学习分布中采样离散架构,每个架构通过选定路径执行生成一个掩码,推理依赖实际执行的子网而非完整候选库。该方法支持架构溯源,每项输出对应特定配置。为避免退化为平均掩码,采用基于交并比的能量距离代理进行集合级监督,匹配一组架构采样预测与标注集合。同时通过进化搜索构建候选库,使随机源支持可优化。所提方法在LIDC-IDRI上实现最先进的分布匹配与假设覆盖,并在两个扩展任务中保持有效性。据我们所知,这是首个将随机分割形式化为学习架构分布,并通过架构采样实现输出多样性的工作。

原文摘要 · Abstract (English)

Stochastic segmentation seeks to represent multiple plausible masks for a single image, which is essential in safety- and quality-critical applications such as medical imaging or building defect inspection. Most existing methods introduce stochasticity by injecting continuous latent variables or by iterative denoising trajectories, whose stochastic sources are difficult to search or audit directly. We propose architecture distributions as a new stochastic source for segmentation: instead of sampling a latent variable or noise, we sample a discrete architecture from a learned distribution over operator choices at multiple searchable positions in a segmentation backbone. Each sampled architecture yields one mask through the selected active path, so inference depends on the executed subnet rather than the complete candidate bank. This approach also supports architectural provenance, since each output corresponds to a specific architecture configuration. To reduce collapse toward averaged masks, we train with set-level supervision by matching a set of architecture-sampled predictions to the annotation set using an IoU-based energy-distance surrogate. We further construct the candidate bank with evolutionary search, making the support of the stochastic source optimizable before distribution learning. The proposed method achieves state-of-the-art distribution matching and hypothesis coverage on LIDC-IDRI, and remains effective on two extension tasks. To the best of our knowledge, this is the first work to formulate stochastic segmentation as learning an architecture distribution and realizing output diversity through architecture sampling.

随机分割架构搜索医学影像可解释性

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