International Association for Cryptologic Research

International Association
for Cryptologic Research

CryptoDB

Reinforcement Learning for Hyperparameter Tuning in Deep Learning-based Side-channel Analysis

Authors:
Jorai Rijsdijk , Delft University of Technology, The Netherlands
Lichao Wu , Delft University of Technology, The Netherlands
Guilherme Perin , Delft University of Technology, The Netherlands
Stjepan Picek , Delft University of Technology, The Netherlands
Download:
DOI: 10.46586/tches.v2021.i3.677-707
URL: https://tches.iacr.org/index.php/TCHES/article/view/8989
Search ePrint
Search Google
Abstract: Deep learning represents a powerful set of techniques for profiling sidechannel analysis. The results in the last few years show that neural network architectures like multilayer perceptron and convolutional neural networks give strong attack performance where it is possible to break targets protected with various countermeasures. Considering that deep learning techniques commonly have a plethora of hyperparameters to tune, it is clear that such top attack results can come with a high price in preparing the attack. This is especially problematic as the side-channel community commonly uses random search or grid search techniques to look for the best hyperparameters.In this paper, we propose to use reinforcement learning to tune the convolutional neural network hyperparameters. In our framework, we investigate the Q-Learning paradigm and develop two reward functions that use side-channel metrics. We mount an investigation on three commonly used datasets and two leakage models where the results show that reinforcement learning can find convolutional neural networks exhibiting top performance while having small numbers of trainable parameters. We note that our approach is automated and can be easily adapted to different datasets. Several of our newly developed architectures outperform the current state-of-the-art results. Finally, we make our source code publicly available. https://github.com/AISyLab/Reinforcement-Learning-for-SCA
Video from TCHES 2021
BibTeX
@article{tches-2021-31298,
  title={Reinforcement Learning for Hyperparameter Tuning in Deep Learning-based Side-channel Analysis},
  journal={IACR Transactions on Cryptographic Hardware and Embedded Systems},
  publisher={Ruhr-Universität Bochum},
  volume={2021, Issue 3},
  pages={677-707},
  url={https://tches.iacr.org/index.php/TCHES/article/view/8989},
  doi={10.46586/tches.v2021.i3.677-707},
  author={Jorai Rijsdijk and Lichao Wu and Guilherme Perin and Stjepan Picek},
  year=2021
}