mlsanityData to 7 Oct 2026

Rule-Based Spatial Mixture-of-Experts U-Net for Explainable Edge Detection

arXiv 2602.05100 ↗First posted 4 Feb 2026cs.CE cs.CV cs.SCcited by 0

On the map: Mixture-of-experts → Vision mixture-of-experts

Abstract (arXiv)

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.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Vision mixture-of-experts, one of 12 directions in Mixture-of-experts. Builds on 7 of its 12 references that are in the map. First posted on arXiv 4 Feb 2026.

Builds on

7 of 12 references.

Closest ancestors · 1

Same direction, newest first

  1. 2021Scaling Vision with Sparse Mixture of Experts10 Jun

Other references · 6

  1. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  2. 2015Holistically-Nested Edge Detectioncited by 590
  3. 2024A Survey on Mixture of Experts in Large Language Modelscited by 233
  4. 2021Pixel Difference Networks for Efficient Edge Detectioncited by 167
  5. 2022Spatial Mixture-of-Expertscited by 14
  6. 2019Adversarial Examples for Edge Detection: They Exist, and They Transfercited by 3

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Edge-Aligned Initialization of Kernels for Steered Mixture-of-Experts
  2. 2024Multilinear Mixture of Experts: Scalable Expert Specialization through Factorization
  3. 2025MoENAS: Mixture-of-Expert based Neural Architecture Search for jointly Accurate, Fair, and Robust Edge Deep Neural Networks
  4. 2026Mixture of Experts with Soft Nearest Neighbor Loss: Resolving Expert Collapse via Representation Disentanglement
  5. 2023Steered Mixture-of-Experts Autoencoder Design for Real-Time Image Modelling and Denoising
  6. 2026Domain-Specialized Object Detection via Model-Level Mixtures of Experts
  7. 2026Mixture-of-Experts Models in Vision: Routing, Optimization, and Generalization
  8. 2024Mixture of Experts in Image Classification: What's the Sweet Spot?