Cryptology ePrint Archive: Report 2021/526

Reinforcement Learning-based Design of Side-channel Countermeasures

Jorai Rijsdijk and Lichao Wu and Guilherme Perin and Stjepan Picek

Abstract: Deep learning represents a powerful set of techniques for profiling side-channel 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.

Category / Keywords: implementation / Side-channel Analysis, Deep Learning, Reinforcement Learning, Reward, Q-policy, Hyperparameter tuning, Convolutional Neural Networks

Date: received 21 Apr 2021

Contact author: picek stjepan at gmail com,jorai@jrijsdijk com,lichao wu9@gmail com,guilhermeperin7@gmail com

Available format(s): PDF | BibTeX Citation

Version: 20210423:122521 (All versions of this report)

Short URL: ia.cr/2021/526


[ Cryptology ePrint archive ]