Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Preprint |
| Published: |
2025
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2512.20885 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866912907368857600 |
|---|---|
| author | Marouani, Sami Singh, Kamal Jeudy, Baptiste Habrard, Amaury |
| author_facet | Marouani, Sami Singh, Kamal Jeudy, Baptiste Habrard, Amaury |
| contents | Accurate prediction of flow delay is essential for optimizing and managing modern communication networks. We investigate three levels of modeling for this task. First, we implement a heterogeneous GNN with attention-based message passing, establishing a strong neural baseline. Second, we propose FlowKANet in which Kolmogorov-Arnold Networks replace standard MLP layers, reducing trainable parameters while maintaining competitive predictive performance. FlowKANet integrates KAMP-Attn (Kolmogorov-Arnold Message Passing with Attention), embedding KAN operators directly into message-passing and attention computation. Finally, we distill the model into symbolic surrogate models using block-wise regression, producing closed-form equations that eliminate trainable weights while preserving graph-structured dependencies. The results show that KAN layers provide a favorable trade-off between efficiency and accuracy and that symbolic surrogates emphasize the potential for lightweight deployment and enhanced transparency. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2512_20885 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | From GNNs to Symbolic Surrogates via Kolmogorov-Arnold Networks for Delay Prediction Marouani, Sami Singh, Kamal Jeudy, Baptiste Habrard, Amaury Machine Learning Networking and Internet Architecture Accurate prediction of flow delay is essential for optimizing and managing modern communication networks. We investigate three levels of modeling for this task. First, we implement a heterogeneous GNN with attention-based message passing, establishing a strong neural baseline. Second, we propose FlowKANet in which Kolmogorov-Arnold Networks replace standard MLP layers, reducing trainable parameters while maintaining competitive predictive performance. FlowKANet integrates KAMP-Attn (Kolmogorov-Arnold Message Passing with Attention), embedding KAN operators directly into message-passing and attention computation. Finally, we distill the model into symbolic surrogate models using block-wise regression, producing closed-form equations that eliminate trainable weights while preserving graph-structured dependencies. The results show that KAN layers provide a favorable trade-off between efficiency and accuracy and that symbolic surrogates emphasize the potential for lightweight deployment and enhanced transparency. |
| title | From GNNs to Symbolic Surrogates via Kolmogorov-Arnold Networks for Delay Prediction |
| topic | Machine Learning Networking and Internet Architecture |
| url | https://arxiv.org/abs/2512.20885 |