Machine Learning-Based Rowhammer Mitigation: A Breakthrough in Hardware Security

Significance 

For cybersecurity, vulnerabilities continue to evolve, and one such vulnerability that has gained significant attention is Rowhammer. Rowhammer is a hardware reliability concern that arises when an attacker repeatedly accesses specific DRAM (Dynamic Random-Access Memory) rows to cause unauthorized changes in physically adjacent memory rows. This vulnerability become a critical concern in the field of computer security due to its potential to enable various attacks, including privilege escalation, sandbox escapes, and breaking cloud isolation. The severity of the Rowhammer problem lies in its exploitation of fundamental DRAM circuit behavior, where bit flips occur when a specific DRAM row is repeatedly activated and precharged, resulting in electromagnetic interference between the aggressor row and its neighboring rows, causing cell capacitors in victim rows to leak faster than normal operation and eventually leading to bit flips. Rowhammer is not confined to specific DRAM generations; it affects both DDR3 and DDR4 DRAM memories making it a widespread and persistent concern across various hardware platforms. Over the years, numerous Rowhammer mitigation techniques have been proposed, both at the hardware and software levels. However, existing mitigation schemes have proven to be either ineffective against emerging Rowhammer attacks or have incurred substantial implementation overhead.

The research study recently published in the peer-reviewed Journal IEEE Transactions on Computer-Aided Design of Integrated Circuits and Systems, led by Dr. Tyler K. Bletsch from Duke University, Dr. Biresh Kumar Joardar, and Professor Krishnendu Chakrabarty from Arizona State University, marks a significant advancement in the ongoing battle against Rowhammer. The authors developed a machine learning (ML)-based technique designed to provide fast and efficient detection of various Rowhammer attacks while maintaining low hardware overhead. They proposed  a solution which leverages machine learning models to analyze memory access patterns and distinguish between benign memory accesses and malicious Rowhammer attacks. By using this ML approach, the authors have successfully developed a Rowhammer mitigation solution that offers several notable advantages. First, the ML model can accurately detect various Rowhammer attacks, including the recently proposed TRRespass, Half-double, and Blacksmith attacks, which have posed challenges to existing mitigation techniques. Secondly, the proposed solution is implemented entirely on-chip, which means it does not require significant external hardware resources, making it a practical and cost-effective choice for mitigating Rowhammer attacks. Moreover, the experimental analysis demonstrated that the ML method introduces significantly lower power and area overhead compared to existing mitigation techniques like Graphene and Blockhammer, with 5% and 19% lower power overhead, respectively.

The researchers’ experimental evaluation of the ML-based Rowhammer mitigation technique provided compelling evidence of its effectiveness. The researchers conducted their experiments using the Gem5 simulator, which emulates a four-core system with various DRAM configurations. They employed a diverse set of applications, including benign applications and Rowhammer attacks, to evaluate the proposed solution’s performance and accuracy.  The experiments revealed that Rowhammer attacks were successful on commercial DDR4 DRAMs, even with Targeted Row Refresh (TRR) enabled. This highlights the urgent need for more effective mitigation techniques.

A critical aspect of the study was the creation of a diverse dataset that included both benign memory access patterns and Rowhammer attacks. Memory-access traces were collected using cycle-accurate simulations on the Gem5 platform. The ML model demonstrated high accuracy in detecting Rowhammer attacks while introducing minimal performance overhead. It outperformed existing techniques in terms of area and power overhead, making it a practical choice for implementation. Moreover, the ML solution exhibited robust performance even when HCfirst (the number of hammers required to cause a bit flip) was as low as 10K. This suggests that the model can adapt to future reductions in HCfirst, which is expected as DRAM technology advances.

The authors’ findings represent a significant step forward in addressing the persistent threat posed by Rowhammer attacks. By harnessing the power of machine learning, the researchers have developed a practical and efficient mitigation technique that not only detects various Rowhammer attacks but also minimizes power and area overhead. The scalability of this solution ensures its relevance in future hardware development.

Machine Learning-Based Rowhammer Mitigation: A Breakthrough in Hardware Security - Advances in Engineering

About the author

Tyler K Bletsch
Assistant Professor of the Practice in the Department of Electrical and Computer Engineering

Tyler joined the faculty at Duke in November 2015 after several years of work in industry with NetApp. He received his Ph.D. from NC State University in 2011, with a research focus on software security. In addition to his work at Duke, he has often been a mentor to FIRST robotics teams. His current professional interests include hardware and software security, robotics, and technology education with an emphasis on project-oriented learning.

About the author

Krishnendu Chakrabarty
Fulton Professor of Microelectronics, School of Electrical, Computer and Energy Engineering
Arizona State University

Krishnendu Chakrabarty (Fellow, IEEE) received the B.Tech. degree from the Indian Institute of Technology Kharagpur, Kharagpur, India, in 1990, and the M.S.E. and Ph.D. degrees from the University of Michigan, Ann Arbor, MI, USA, in 1992 and 1995, respectively.

His current research projects include: design-for-testability of 2.5D/3-D integrated circuits and heterogeneous integration; hardware security; AI accelerators; microfluidic biochips; AI for healthcare; and neuromorphic computing systems.

Prof. Chakrabarty is a recipient of the National Science Foundation CAREER Award, the Office of Naval Research Young Investigator Award, the Humboldt Research Award from the Alexander von Humboldt Foundation, Germany, the IEEE Transactions on CAD Donald O. Pederson Best Paper Award in 2015, the IEEE Transactions on Very Large Scale Integration (VLSI) Systems Prize Paper Award in 2021, the ACM Transactions on Design Automation of Electronic Systems Best Paper Award in 2017, multiple IBM Faculty Awards and HP Labs Open Innovation Research Awards, and over a dozen best paper awards at major conferences. He is also a recipient of the IEEE Computer Society Technical Achievement Award in 2015, the IEEE Circuits and Systems Society Charles A. Desoer Technical Achievement Award in 2017, the IEEE Circuits and Systems Society Vitold Belevitch Award in 2021, the Semiconductor Research Corporation Technical Excellence Award in 2018, the Semiconductor Research Corporation Aristotle Award in 2022, the IEEE-HKN Asad M. Madni Outstanding Technical Achievement and Excellence Award in 2021, and the IEEE Test Technology Technical Council Bob Madge Innovation Award in 2018.

Reference

Biresh Kumar Joardar; Tyler K. Bletsch; Krishnendu Chakrabarty. Machine Learning-Based Rowhammer Mitigation IEEE Transactions on Computer-Aided Design of Integrated Circuits and Systems ( Volume: 42, Issue: 5,   2023)

Go to IEEE Transactions on Computer-Aided Design of Integrated Circuits and Systems

Check Also

Non-Volatile Memory and Neuro-Inspired Computing: Machine Learning-Powered Simulations of Phase-Change Materials - Advances in Engineering

Non-Volatile Memory and Neuro-Inspired Computing: Machine Learning-Powered Simulations of Phase-Change Materials