arXiv subjects

Dehao Kong

Publications and source records attributed to Dehao Kong.

4 recordsLinked to original sources

arXiv2025

Bayesian Reasoning Enabled by Spin-Orbit Torque Magnetic Tunnel Junctions

Bayesian networks play an increasingly important role in data mining, inference, and reasoning with the rapid development of artificial intelligence. In this paper, we present proof-of-concept experiments demonstrating the use of spin-orbit torque magnetic tunnel junctions (SOT-MTJs) in Bayesian network reasoning. Not only can the target probability distribution function (PDF) of a Bayesian network be precisely formulated by a conditional probability table as usual but also quantitatively parameterized by a probabilistic forward propagating neuron network. Moreover, the parameters of the network can also approach the optimum through a simple point-by point training algorithm, by leveraging which we do not need to memorize all historical data nor statistically summarize conditional probabilities behind them, significantly improving storage efficiency and economizing data pretreatment. Furthermore, we developed a simple medical diagnostic system using the SOT-MTJ as a random number generator and sampler, showcasing the application of SOT-MTJ-based Bayesian reasoning. This SOT-MTJ-based Bayesian reasoning shows great promise in the field of artificial probabilistic neural network, broadening the scope of spintronic device applications and providing an efficient and low-storage solution for complex reasoning tasks.

physics.app-ph↗

arXiv2025

Engineering-Oriented Design of Drift-Resilient MTJ Random Number Generator via Hybrid Control Strategies

Magnetic Tunnel Junctions (MTJs) have shown great promise as hardware sources for true random number generation (TRNG) due to their intrinsic stochastic switching behavior. However, practical deployment remains challenged by drift in switching probability caused by thermal fluctuations, device aging, and environmental instability. This work presents an engineering-oriented, drift-resilient MTJ-based TRNG architecture, enabled by a hybrid control strategy that combines self-stabilizing feedback with pulse width modulation. A key component is the Downcalibration-2 scheme, which updates the control parameter every two steps using only integer-resolution timing, ensuring excellent statistical quality without requiring bit discarding, pre-characterization, or external calibration. Extensive experimental measurements and numerical simulations demonstrate that this approach maintains stable randomness under dynamic temperature drift, using only simple digital logic. The proposed architecture offers high throughput, robustness, and scalability, making it well-suited for secure hardware applications, embedded systems, and edge computing environments.

physics.app-ph↗

arXiv2025

Probabilistic Greedy Algorithm Solver Using Magnetic Tunneling Junctions for Traveling Salesman Problem

Combinatorial optimization problems are foundational challenges in fields such as artificial intelligence, logistics, and network design. Traditional algorithms, including greedy methods and dynamic programming, often struggle to balance computational efficiency and solution quality, particularly as problem complexity scales. To overcome these limitations, we propose a novel and efficient probabilistic optimization framework that integrates true random number generators (TRNGs) based on spin-transfer torque magnetic tunneling junctions (STT-MTJs). The inherent stochastic switching behavior of STT-MTJs enables dynamic configurability of random number distributions, which we leverage to introduce controlled randomness into a probabilistic greedy algorithm. By tuning a temperature parameter, our algorithm seamlessly transitions between deterministic and stochastic strategies, effectively balancing exploration and exploitation. Furthermore, we apply this framework to the traveling salesman problem (TSP), showcasing its ability to consistently produce high-quality solutions across diverse problem scales. Our algorithm demonstrates superior performance in both solution quality and convergence speed compared to classical approaches, such as simulated annealing and genetic algorithms. Specifically, in larger TSP instances involving up to 70 cities, it retains its performance advantage, achieving near-optimal solutions with fewer iterations and reduced computational costs. This work highlights the potential of integrating MTJ-based TRNGs into optimization algorithms, paving the way for future applications in probabilistic computing and hardware-accelerated optimization.

physics.app-ph↗

arXiv2024

PALM: A Efficient Performance Simulator for Tiled Accelerators with Large-scale Model Training

Deep learning (DL) models are piquing high interest and scaling at an unprecedented rate. To this end, a handful of tiled accelerators have been proposed to support such large-scale training tasks. However, these accelerators often incorporate numerous cores or tiles even extending to wafer-scale, substantial on-chip bandwidth, and distributed memory systems. This results in an exceedingly complex design space. Moreover, conducting actual training experiments to find optimal configurations is impractical due to time constraints. Hence, predicting the optimal mapping of various parallelisms to such tiled system architectures becomes crucial. In this study, leveraging an analysis of existing mainstream DL model training strategies, we introduce a performance simulator named PALM. PALM targets both the training and inference processes for tiled accelerators, aiming to inspire the design of current and future accelerators. Specifically, (i) we establish a scheduling mechanism among tiled accelerators based on an event-driven framework; (ii) we support user-configurable pipeline, tensor, and data parallelism on tiled accelerators, determining the absolute performance throughput under these parallelism strategies; (iii) we model the interaction of on-chip SRAM, NoC, and off-chip DRAM during operator execution. This work is available here: https://github.com/fangjh21/PALM.

cs.DC↗