Toward Intrinsically Interpretable AI in Optical Networks Using KAN-based Symbolic Regression

Artificial intelligence (AI) and machine learning (ML) have demonstrated remarkable performance in addressing multiple optical network tasks. However, the lack of interpretability and transparency in black-box models impedes understanding of model behaviors and thereby leads to subpar generalization ability and hinders operators from trusting model decisions. In this paper, we target building intrinsically interpretable AI models for optical networks using Kolmogorov–Arnold network (KAN)-based symbolic regression (SR). Specifically, KAN-based SR parameterizes intrinsically interpretable formulas by learnable activation functions (i.e., spline functions) and approximates the true mappings by formula fitting and pruning. We perform a case study on KAN-based SR for QoT estimation and demonstrate successful learning of interpretable formulas via a fully transparent training process. The results also show superior generalization ability of the proposed design compared with the baselines, achieving a mean squared error (MSE) of below 0.035 dB for optical signal-to-noise ratio (OSNR) prediction under different distributions of training and testing data.

Authors

Institutions

Publication Details

Published
2024-10-15
DOI
https://doi.org/10.1109/fnwf63303.2024.11028761
Primary Topic
Neural Networks and Reservoir Computing
Type
article
Field-Weighted Citation Impact
0.00

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Toward Intrinsically Interpretable AI in Optical Networks Using KAN-based Symbolic Regression

Hanyu Gao, Aoxue Wang, Zhaohui Li, Xiaoliang Chen et al.
Neural Networks and Reservoir Computing
article

Toward Intrinsically Interpretable AI in Optical Networks Using KAN-based Symbolic Regression

Hanyu Gao, Aoxue Wang, Zhaohui Li, Xiaoliang Chen, Zhenlin Ouyang
article en

Abstract

Artificial intelligence (AI) and machine learning (ML) have demonstrated remarkable performance in addressing multiple optical network tasks. However, the lack of interpretability and transparency in black-box models impedes understanding of model behaviors and thereby leads to subpar generalization ability and hinders operators from trusting model decisions. In this paper, we target building intrinsically interpretable AI models for optical networks using Kolmogorov–Arnold network (KAN)-based symbolic regression (SR). Specifically, KAN-based SR parameterizes intrinsically interpretable formulas by learnable activation functions (i.e., spline functions) and approximates the true mappings by formula fitting and pruning. We perform a case study on KAN-based SR for QoT estimation and demonstrate successful learning of interpretable formulas via a fully transparent training process. The results also show superior generalization ability of the proposed design compared with the baselines, achieving a mean squared error (MSE) of below 0.035 dB for optical signal-to-noise ratio (OSNR) prediction under different distributions of training and testing data.

Sun Yat-sen University (CN)
National Natural Science Foundation of China, National Key Research and Development Program of China
Reduced inequalities
Openalex Percentile: Top 72%
Neural Networks and Reservoir Computing
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.