SearcharxivSearch

arXiv subjects

Cheng-Kai Wang

Publications and source records attributed to Cheng-Kai Wang.

2 recordsLinked to original sources

SCPL: Enhancing Neural Network Training Throughput with Decoupled Local Losses and Model Parallelism

Adopting large-scale AI models in enterprise information systems is often hindered by high training costs and long development cycles, posing a significant managerial challenge. The standard end-to-end backpropagation (BP) algorithm is a primary driver of modern AI, but it is also the source of inefficiency in training deep networks. This paper introduces a new training methodology, Supervised Contrastive Parallel Learning (SCPL), that addresses this issue by decoupling BP and transforming a long gradient flow into multiple short ones. This design enables the simultaneous computation of parameter gradients in different layers, achieving superior model parallelism and enhancing training throughput. Detailed experiments are presented to demonstrate the efficiency and effectiveness of our model compared to BP, Early Exit, GPipe, and Associated Learning (AL), a state-of-the-art method for decoupling backpropagation. By mitigating a fundamental performance bottleneck, SCPL provides a practical pathway for organizations to develop and deploy advanced information systems more cost-effectively and with greater agility. The experimental code is released for reproducibility. https://github.com/minyaho/scpl/

cs.LG

Experimental quantum stochastic walks simulating associative memory of Hopfield neural networks

With the increasing crossover between quantum information and machine learning, quantum simulation of neural networks has drawn unprecedentedly strong attention, especially for the simulation of associative memory in Hopfield neural networks due to their wide applications and relatively simple structures that allow for easier mapping to the quantum regime. Quantum stochastic walk, a strikingly powerful tool to analyze quantum dynamics, has been recently proposed to simulate the firing pattern and associative memory with a dependence on Hamming Distance. We successfully map the theoretical scheme into a three-dimensional photonic quantum chip and realize quantum stochastic walk evolution through well-controlled detunings of the propagation constant. We demonstrate a good match rate of the associative memory between the experimental quantum scheme and the expected result for Hopfield neural networks. The ability of quantum simulation for an important feature of a neural network, combined with the scalability of our approach through low-loss integrated chip and straightforward Hamiltonian engineering, provides a primary but steady step towards photonic artificial intelligence devices for optimization and computation tasks of greatly improved efficiencies.

quant-ph