This repository contains various components related to the research paper and source codes of the multimodal hardware Trojan detection framework.
Rahul Vishwakarma & Amin Rezaei
The risk of hardware Trojans being inserted at various stages of chip production has increased in a zero-trust fabless era. To counter this, various machine learning solutions have been developed for the detection of hardware Trojans. While most of the focus has been on either a statistical or deep learning approach, the limited number of Trojan-infected benchmarks affects the detection accuracy and restricts the possibility of detecting zero-day Trojans. To close the gap, we first employ generative adversarial networks to amplify our data in two alternative representation modalities: a graph and a tabular, which ensure a representative distribution of the dataset. Further, we propose a multimodal deep learning approach to detect hardware Trojans and evaluate the results from both early fusion and late fusion strategies. We also estimate the uncertainty quantification metrics of each prediction for risk-aware decision-making. The results not only validate the effectiveness of our suggested hardware Trojan detection technique but also pave the way for future studies utilizing multimodality and uncertainty quantification to tackle other hardware security problems.
We have used two different dataset for the detection of hardware Trojans.
The dataste has binary label: Trojan Free and Trojan Infected.
Graph dataset are represented as .json files and Tabular dataaset in .csv file
The source code and results are in folder: synthetic-multimodal-graph_and_table
The dataste has binary label: Trojan Free and Trojan Infected.
https://github.com/AICPS/hw2vec.
The source code and results are in the folder: source
Dataset Source: https://trust-hub.org/#/benchmarks/chip-level-trojan
Use the below Python source code for reproducing the results in the paper.
This provides the ROC curve.
source/01_table.py
Provides the the ROC curve plots.
source/02_graph.py
Provides the brier score, accuracy, and confusion matrix.
source/03_early_fusion.py
Provides the brier score, accuracy, and confusion matrix.
source/04_late_fusion.py
- Proposing a multimodal learning approach using graph and euclidean data of the hardware circuits.
- Suggesting a model fusion approach using p-values with uncertainty quantifier.
- Addressing the critical issue of missing modalities and small dataset.
GNU General Public License v3.0
Journal Article:
@ARTICLE{Journal-2025,
author={Vishwakarma, Rahul and Rezaei, Amin},
title={Uncertainty-Aware Unimodal and Multimodal Learning for Evolving Hardware Trojan Detection},
journal={Journal of Hardware and Systems Security},
year={2025},
volume={},
number={},
pages={},
doi={10.1007/s41635-025-00160-2}
}
Conference Article:
@INPROCEEDINGS{Conference-2024,
author={Vishwakarma, Rahul and Rezaei, Amin},
booktitle={2024 Design, Automation & Test in Europe Conference & Exhibition (DATE)},
title={Uncertainty-Aware Hardware Trojan Detection Using Multimodal Deep Learning},
year={2024},
volume={},
number={},
pages={1-6},
doi={10.23919/DATE58400.2024.10546558}
}
