Generalized few-shot transfer learning architecture for modeling the EDFA gain spectrum

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Raj, Agastya, Wang, Zehao, Chen, Tingjun, Kilper, Daniel C, Ruffini, Marco
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911081716252672
author Raj, Agastya
Wang, Zehao
Chen, Tingjun
Kilper, Daniel C
Ruffini, Marco
author_facet Raj, Agastya
Wang, Zehao
Chen, Tingjun
Kilper, Daniel C
Ruffini, Marco
contents Accurate modeling of the gain spectrum in Erbium-Doped Fiber Amplifiers (EDFAs) is essential for optimizing optical network performance, particularly as networks evolve toward multi-vendor solutions. In this work, we propose a generalized few-shot transfer learning architecture based on a Semi-Supervised Self-Normalizing Neural Network (SS-NN) that leverages internal EDFA features - such as VOA input or output power and attenuation, to improve gain spectrum prediction. Our SS-NN model employs a two-phase training strategy comprising unsupervised pre-training with noise-augmented measurements and supervised fine-tuning with a custom weighted MSE loss. Furthermore, we extend the framework with transfer learning (TL) techniques that enable both homogeneous (same-feature space) and heterogeneous (different-feature sets) model adaptation across booster, preamplifier, and ILA EDFAs. To address feature mismatches in heterogeneous TL, we incorporate a covariance matching loss to align second-order feature statistics between source and target domains. Extensive experiments conducted across 26 EDFAs in the COSMOS and Open Ireland testbeds demonstrate that the proposed approach significantly reduces the number of measurements requirements on the system while achieving lower mean absolute errors and improved error distributions compared to benchmark methods.
format Preprint
id arxiv_https___arxiv_org_abs_2507_21728
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Generalized few-shot transfer learning architecture for modeling the EDFA gain spectrum
Raj, Agastya
Wang, Zehao
Chen, Tingjun
Kilper, Daniel C
Ruffini, Marco
Networking and Internet Architecture
Machine Learning
Accurate modeling of the gain spectrum in Erbium-Doped Fiber Amplifiers (EDFAs) is essential for optimizing optical network performance, particularly as networks evolve toward multi-vendor solutions. In this work, we propose a generalized few-shot transfer learning architecture based on a Semi-Supervised Self-Normalizing Neural Network (SS-NN) that leverages internal EDFA features - such as VOA input or output power and attenuation, to improve gain spectrum prediction. Our SS-NN model employs a two-phase training strategy comprising unsupervised pre-training with noise-augmented measurements and supervised fine-tuning with a custom weighted MSE loss. Furthermore, we extend the framework with transfer learning (TL) techniques that enable both homogeneous (same-feature space) and heterogeneous (different-feature sets) model adaptation across booster, preamplifier, and ILA EDFAs. To address feature mismatches in heterogeneous TL, we incorporate a covariance matching loss to align second-order feature statistics between source and target domains. Extensive experiments conducted across 26 EDFAs in the COSMOS and Open Ireland testbeds demonstrate that the proposed approach significantly reduces the number of measurements requirements on the system while achieving lower mean absolute errors and improved error distributions compared to benchmark methods.
title Generalized few-shot transfer learning architecture for modeling the EDFA gain spectrum
topic Networking and Internet Architecture
Machine Learning
url https://arxiv.org/abs/2507.21728