Accelerating the pace of machine learning

You are here

ECE’s Rick Blum seeks to remove the bottleneck in wireless, distributed learning applications

Machine learning happens a lot like erosion.

Data is hurled at a mathematical model like grains of sand skittering across a rocky landscape. Some of those grains simply sail along with little or no impact. But some of them make their mark: testing, hardening, and ultimately reshaping the landscape according to inherent patterns and fluctuations that emerge over time.

Effective? Yes. Efficient? Not so much.

Rick Blum, the Robert W. Wieseman Professor of Electrical and Computer Engineering at Lehigh University, seeks to bring efficiency to distributed learning techniques emerging as crucial to modern artificial intelligence (AI) and machine learning (ML). In essence, his goal is to hurl far fewer grains of data without degrading the overall impact.

In the paper "Distributed Learning With Sparsified Gradient Differences," published in a special ML-focused issue of the IEEE Journal of Selected Topics in Signal Processing, Blum and collaborators propose the use of "Gradient Descent method with Sparsification and Error Correction," or GD-SEC, to improve the communications efficiency of machine learning conducted in a "worker-server" wireless architecture. The issue was published May 17, 2022.

"Problems in distributed optimization appear in various scenarios that typically rely on wireless communications," he says. "Latency, scalability, and privacy are fundamental challenges."

"Various distributed optimization algorithms have been developed to solve this problem," he continues,"and one primary method is to employ classical GD in a worker-server architecture. In this environment, the central server updates the model's parameters after aggregating data received from all workers, and then broadcasts the updated parameters back to the workers. But the overall performance is limited by the fact that each worker must transmit all of its data all of the time. When training a deep neural network, this can be on the order of 200 MB from each worker device at each iteration. This communication step can easily become a significant bottleneck on overall performance, especially in federated learning and edge AI systems."

Through the use of GD-SEC, Blum explains, communication requirements are significantly reduced. The technique employs a data compression approach where each worker sets small magnitude gradient components to zero—the signal-processing equivalent of not sweating the small stuff. The worker then only transmits to the server the remaining non-zero components. In other words, meaningful, usable data are the only packets launched at the model.

"Current methods create a situation where each worker has expensive computational cost; GD-SEC is relatively cheap where only one GD step is needed at each round," says Blum.

Professor Blum's collaborators on this project include his former student Yicheng Chen '19G '21PhD, now a software engineer with LinkedIn; Martin Takáč, an associate professor at the Mohamed bin Zayed University of Artificial Intelligence; and Brian M. Sadler, a Life Fellow of the IEEE, U.S. Army Senior Scientist for Intelligent Systems, and Fellow of the Army Research Laboratory.

The article, "Distributed Learning With Sparsified Gradient Differences," appears in the April 2022 special issue of the IEEE Journal of Selected Topics in Signal Processing.

About Professor Blum
Prof. Rick Blum holds the Robert W. Wieseman Endowed Professorship in Electrical Engineering. He served as the lead Lehigh PI for its DoE Cybersecurity Center (SEEDs) and is director of its Signal Processing and Communication Research Lab.

His group contributes to the foundational theory of machine learning and statistical decision making while illustrating this theory and the related algorithms by employing the application areas of cybersecurity, internet of things, cyberphysical systems, sensor networking, energy networks, communications, radar, and sensor processing.

Prior to joining the Lehigh faculty, he was a senior member of technical staff at General Electric Aerospace and graduated from GE's Advanced Course in Engineering. He served on the editorial board for the Journal of Advances in Information Fusion of the International Society of Information Fusion. He was an associate editor for IEEE Transactions on Signal Processing and for IEEE Communications Letters. He has edited special issues for IEEE Transactions on Signal Processing, IEEE Journal of Selected Topics in Signal Processing and IEEE Journal on Selected Areas in Communications. He was a member of the SAM Technical Committee (TC) of the IEEE Signal Processing Society, the Signal Processing for Communications TC of the IEEE Signal Processing Society, and the Communications Theory TC of the IEEE Communication Society. 

Prof. Rick Blum is an IEEE Fellow, an IEEE Signal Processing Society Distinguished Lecturer, an IEEE Third Millennium Medal winner, and an ONR Young Investigator.

Source: Article from P.C Rossin College of Engineering and Applied Science News (website)
Read Full article >