This repository consists the code for the implementation of U-Net architecture but with Kolmogorov-Arnold Convolutions instead of regular convolutions.

KANs are promising alternatives of Multi-Layer Perceptrons (MLPs). KANs have strong mathematical foundations just like MLPs: MLPs are based on the universal approximation theorem, while KANs are based on Kolmogorov-Arnold representation theorem. KANs and MLPs are dual: KANs have activation functions on edges, while MLPs have activation functions on nodes. KAN seems to be more parameter efficient than MLPs, but each KAN Layer has more parameters than a MLP layer.

For more information about this novel architecture please visit:

• The official Pytorch implementation of the architecture: https://github.com/KindXiaoming/pykan
• The research paper: https://arxiv.org/abs/2404.19756

git clone https://github.com/JaouadT/KANU_Net.git
cd KANU_Net
pip install -r requirements.txt

python train.py --model KANU_Net --dataset BUSI --gpu 0

The following are the results after training KANU_Net and regular U-Net with the same experimental setup:

This model is built upon U-Net and the authors of the implementation of the KA-Conv. We extend our gratitude to the creators of the original KAN for their pioneering work in the field.