학술논문

Imitation Learning-based Equalizer Design Optimization Method on PCIe 6.0
Document Type
Conference
Source
2023 IEEE Electrical Design of Advanced Packaging and Systems (EDAPS) Advanced Packaging and Systems (EDAPS), 2023 IEEE Electrical Design of. :1-3 Dec, 2023
Subject
Components, Circuits, Devices and Systems
Fields, Waves and Electromagnetics
Power, Energy and Industry Applications
Signal Processing and Analysis
Equalizers
Training data
Artificial neural networks
Packaging
Deep reinforcement learning
Bayes methods
Optimization
Bayesian optimization (BO)
Deep neural network (DNN)
Deep reinforcement learning (DRL)
Imitation learning
PAM-4
PCIe 6.0.
Random search (RS)
Language
ISSN
2151-1233
Abstract
In this paper, we proposed imitation learning-based equalizer design optimization method on PCIe 6.0. With each update of PCIe, the equalizer scheme has evolved to compensate for channel loss. However, due to the increasing number of equalizer parameters and their inter-coupling, co-optimizing these parameters becomes essential. While various methods have been researched to optimize equalizer parameters, they often require re-optimization for different conditions or they inefficiently train the DNN policy used for optimization. In contrast, our method utilizes imitation learning scheme. We collected training data offline through data exploration using expert policies like Bayesian optimization (BO) and random search (RS), and subsequently trained the DNN policy to mimic collected data. For validation, we compared the PAM-4 eye opening of our proposed method with that of BO and RS. As a result, the performance of our proposed method, with just a single simulation, surpassed that of both RS and BO, each simulated 100 times.