Paper Reading AI Learner

Bridging KAN and MLP: MJKAN, a Hybrid Architecture with Both Efficiency and Expressiveness

2025-07-07 06:13:32
Hanseon Joo, Hayoung Choi, Ook Lee, Minjong Cheon

Abstract

Kolmogorov-Arnold Networks (KANs) have garnered attention for replacing fixed activation functions with learnable univariate functions, but they exhibit practical limitations, including high computational costs and performance deficits in general classification tasks. In this paper, we propose the Modulation Joint KAN (MJKAN), a novel neural network layer designed to overcome these challenges. MJKAN integrates a FiLM (Feature-wise Linear Modulation)-like mechanism with Radial Basis Function (RBF) activations, creating a hybrid architecture that combines the non-linear expressive power of KANs with the efficiency of Multilayer Perceptrons (MLPs). We empirically validated MJKAN's performance across a diverse set of benchmarks, including function regression, image classification (MNIST, CIFAR-10/100), and natural language processing (AG News, SMS Spam). The results demonstrate that MJKAN achieves superior approximation capabilities in function regression tasks, significantly outperforming MLPs, with performance improving as the number of basis functions increases. Conversely, in image and text classification, its performance was competitive with MLPs but revealed a critical dependency on the number of basis functions. We found that a smaller basis size was crucial for better generalization, highlighting that the model's capacity must be carefully tuned to the complexity of the data to prevent overfitting. In conclusion, MJKAN offers a flexible architecture that inherits the theoretical advantages of KANs while improving computational efficiency and practical viability.

Abstract (translated)

Kolmogorov-Arnold Networks(KAN)因其用可学习的一元函数替代固定激活函数而受到关注,但它们在实际应用中存在一些限制,包括计算成本高和一般分类任务中的性能不足。本文提出了一种新的神经网络层——调制联合KAN(MJKAN),旨在克服这些挑战。MJKAN结合了类似于FiLM(按特征线性调制)的机制和径向基函数(RBF)激活,构建了一个混合架构,它融合了KAN的非线性表达能力和多层感知机(MLP)的效率。 我们通过一系列基准测试对MJKAN的表现进行了实证验证,包括函数回归、图像分类(MNIST, CIFAR-10/100)和自然语言处理(AG News, SMS Spam)。结果表明,在函数回归任务中,MJKAN展示了优越的逼近能力,并显著超越了MLP,且随着基函数数量的增加性能有所提升。然而,在图像和文本分类中,其表现与MLP相当,但显现出对基函数数量的关键依赖性。我们发现较小的基础大小对于更好的泛化至关重要,这表明模型容量需要根据数据复杂度进行仔细调整以防止过拟合。 综上所述,MJKAN提供了一种灵活的架构,继承了KAN理论上的优势,同时在计算效率和实际可行性方面进行了改进。

URL

https://arxiv.org/abs/2507.04690

PDF

https://arxiv.org/pdf/2507.04690.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot