提出可解释的边缘检测模型,让深度学习的决策过程透明化。
Rule-Based Spatial Mixture-of-Experts U-Net for Explainable Edge Detection
- 用空间自适应专家混合模块动态选择上下文或边界特征
- 在BSDS500上达到0.7628的F分数,接近顶尖模型性能
- 通过规则触发图和策略图实现像素级决策解释,适合需要可解释性的场景
U-Net及其变体在边缘检测任务中表现卓越,被全球生成式AI服务广泛使用。然而其决策过程如同黑箱,缺乏透明性,限制了在安全关键领域的应用。为此,本文提出基于规则的空间专家混合U-Net(sMoE U-Net)。该模型引入两项创新:(1) 在解码器跳跃连接中嵌入空间自适应专家混合(sMoE)块,根据局部特征统计动态切换“上下文”(平滑)与“边界”(锐利)专家;(2) 采用Takagi-Sugeno-Kang(TSK)模糊头替代标准分类层,通过显式IF-THEN规则融合深层语义特征与启发式边缘信号。在BSDS500基准测试中,模型取得0.7628的最优数据集尺度(ODS)F分数,接近纯深度基线模型HED的0.7688,优于标准U-Net的0.7437。更重要的是,模型提供像素级可解释性,支持可视化边缘判断是源于强梯度、高语义置信度,还是特定规则组合。
原文摘要 · Abstract (English)
Deep learning models like U-Net and its variants, have established state-of-the-art performance in edge detection tasks and are used by Generative AI services world-wide for their image generation models. However, their decision-making processes remain opaque, operating as "black boxes" that obscure the rationale behind specific boundary predictions. This lack of transparency is a critical barrier in safety-critical applications where verification is mandatory. To bridge the gap between high-performance deep learning and interpretable logic, we propose the Rule-Based Spatial Mixture-of-Experts U-Net (sMoE U-Net). Our architecture introduces two key innovations: (1) Spatially-Adaptive Mixture-of-Experts (sMoE) blocks integrated into the decoder skip connections, which dynamically gate between "Context" (smooth) and "Boundary" (sharp) experts based on local feature statistics; and (2) a Takagi-Sugeno-Kang (TSK) Fuzzy Head that replaces the standard classification layer. This fuzzy head fuses deep semantic features with heuristic edge signals using explicit IF-THEN rules. We evaluate our method on the BSDS500 benchmark, achieving an Optimal Dataset Scale (ODS) F-score of 0.7628, effectively matching purely deep baselines like HED (0.7688) while outperforming the standard U-Net (0.7437). Crucially, our model provides pixel-level explainability through "Rule Firing Maps" and "Strategy Maps," allowing users to visualize whether an edge was detected due to strong gradients, high semantic confidence, or specific logical rule combinations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。