Implicit U-KAN2.0: Dynamic, Efficient and Interpretable Medical Image Segmentation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Cheng, Chun-Wun, Zhao, Yining, Cheng, Yanqi, Montoya-Zegarra, Javier A., Schönlieb, Carola-Bibiane, Aviles-Rivero, Angelica I
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917313099333632
author Cheng, Chun-Wun
Zhao, Yining
Cheng, Yanqi
Montoya-Zegarra, Javier A.
Schönlieb, Carola-Bibiane
Aviles-Rivero, Angelica I
author_facet Cheng, Chun-Wun
Zhao, Yining
Cheng, Yanqi
Montoya-Zegarra, Javier A.
Schönlieb, Carola-Bibiane
Aviles-Rivero, Angelica I
contents Image segmentation is a fundamental task in both image analysis and medical applications. State-of-the-art methods predominantly rely on encoder-decoder architectures with a U-shaped design, commonly referred to as U-Net. Recent advancements integrating transformers and MLPs improve performance but still face key limitations, such as poor interpretability, difficulty handling intrinsic noise, and constrained expressiveness due to discrete layer structures, often lacking a solid theoretical foundation.In this work, we introduce Implicit U-KAN 2.0, a novel U-Net variant that adopts a two-phase encoder-decoder structure. In the SONO phase, we use a second-order neural ordinary differential equation (NODEs), called the SONO block, for a more efficient, expressive, and theoretically grounded modeling approach. In the SONO-MultiKAN phase, we integrate the second-order NODEs and MultiKAN layer as the core computational block to enhance interpretability and representation power. Our contributions are threefold. First, U-KAN 2.0 is an implicit deep neural network incorporating MultiKAN and second order NODEs, improving interpretability and performance while reducing computational costs. Second, we provide a theoretical analysis demonstrating that the approximation ability of the MultiKAN block is independent of the input dimension. Third, we conduct extensive experiments on a variety of 2D and a single 3D dataset, demonstrating that our model consistently outperforms existing segmentation networks. Project Website: https://math-ml-x.github.io/IUKAN2/
format Preprint
id arxiv_https___arxiv_org_abs_2503_03141
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Implicit U-KAN2.0: Dynamic, Efficient and Interpretable Medical Image Segmentation
Cheng, Chun-Wun
Zhao, Yining
Cheng, Yanqi
Montoya-Zegarra, Javier A.
Schönlieb, Carola-Bibiane
Aviles-Rivero, Angelica I
Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
Image segmentation is a fundamental task in both image analysis and medical applications. State-of-the-art methods predominantly rely on encoder-decoder architectures with a U-shaped design, commonly referred to as U-Net. Recent advancements integrating transformers and MLPs improve performance but still face key limitations, such as poor interpretability, difficulty handling intrinsic noise, and constrained expressiveness due to discrete layer structures, often lacking a solid theoretical foundation.In this work, we introduce Implicit U-KAN 2.0, a novel U-Net variant that adopts a two-phase encoder-decoder structure. In the SONO phase, we use a second-order neural ordinary differential equation (NODEs), called the SONO block, for a more efficient, expressive, and theoretically grounded modeling approach. In the SONO-MultiKAN phase, we integrate the second-order NODEs and MultiKAN layer as the core computational block to enhance interpretability and representation power. Our contributions are threefold. First, U-KAN 2.0 is an implicit deep neural network incorporating MultiKAN and second order NODEs, improving interpretability and performance while reducing computational costs. Second, we provide a theoretical analysis demonstrating that the approximation ability of the MultiKAN block is independent of the input dimension. Third, we conduct extensive experiments on a variety of 2D and a single 3D dataset, demonstrating that our model consistently outperforms existing segmentation networks. Project Website: https://math-ml-x.github.io/IUKAN2/
title Implicit U-KAN2.0: Dynamic, Efficient and Interpretable Medical Image Segmentation
topic Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2503.03141