RecSys

Architectural Support for Deep Recommendation Systems(RecSys)

RecSys Figure 1

Recommendation systems form the backbone of popular internet services like entertainment streaming, e-commerce and social media (e.g., Netflix, Amazon, Facebook). These deep learning-based systems not only present unique compute challenges compared to well-studied DNNs but also introduce significant infrastructure demands for at-scale deployment. In order to make recommendation more efficient, solutions will have to integrate insights across the entire execution stack (as shown below).

At the use-case level, our group looks at optimizing both training and inference cycles with datacenter-scale (e.g., workload scheduling) as well as mobile-centric (e.g., privacy preservation) solutions. Framing the problem in these specific contexts allows us to propose algorithmic adjustments (e.g., model compression and partitioning) that are most appropriate for each use case. Last but not least, we quantify the implications of current heterogeneous hardware designs on recommendation workloads and use these insights to propose future architectures specialized for recommendation.

Select Publications

2021

Mark Wilkening, Udit Gupta, Samuel Hsia, Caroline Trippel, Carole-Jean Wu, David Brooks, and Gu-Yeon Wei. 2021. “RecSSD: Near Data Processing for Solid State Drive Based Recommendation Inference”. ASPLOS 2021: Proceedings of the 26th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Pp. 717–729
Mark Wilkening, Udit Gupta, Samuel Hsia, Caroline Trippel, Carole-Jean Wu, David Brooks, and Gu-Yeon Wei. 2021. “RecSSD: Near Data Processing for Solid State Drive Based Recommendation Inference”. ASPLOS 2021: Proceedings of the 26th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Pp. 717–729
Udit Gupta, Samuel Hsia, Jeff Zhang, Mark Wilkening, Javin Pombra, Hsien-Hsin S. Lee, Gu-Yeon Wei, Carole-Jean Wu, and David Brooks. 2021. “RecPipe: Co-Designing Models and Hardware to Jointly Optimize Recommendation Quality and Performance”. MICRO ’21: MICRO-54: 54th Annual IEEE ACM International Symposium on Microarchitecture, Pp. 870–884
Udit Gupta, Samuel Hsia, Jeff Zhang, Mark Wilkening, Javin Pombra, Hsien-Hsin S. Lee, Gu-Yeon Wei, Carole-Jean Wu, and David Brooks. 2021. “RecPipe: Co-Designing Models and Hardware to Jointly Optimize Recommendation Quality and Performance”. MICRO ’21: MICRO-54: 54th Annual IEEE ACM International Symposium on Microarchitecture, Pp. 870–884

2020

Samuel Hsia, Udit Gupta, Wilkening Mark, Carole Wu, Gu-Yeon Wei, and David Brooks. 2020. “Cross-Stack Workload Characterization of Deep Recommendation Systems”. In 2020 IEEE International Symposium on Workload Characterization (IISWC)
Samuel Hsia, Udit Gupta, Wilkening Mark, Carole Wu, Gu-Yeon Wei, and David Brooks. 2020. “Cross-Stack Workload Characterization of Deep Recommendation Systems”. In 2020 IEEE International Symposium on Workload Characterization (IISWC)
Udit Gupta, Samuel Hsia, Vikram Saraph, Xiaodong Wang, Brandon Reagen, Gu-Yeon Wei, Hsien-Hsin S. Lee, Carole-Jean Wu, and David Brooks. 2020. “DeepRecSys: A System for Optimizing End-To-End At-Scale Neural Recommendation Inference”. In . The 47th IEEE/ACM International Symposium on Computer Architecture (ISCA 2020)
Udit Gupta, Samuel Hsia, Vikram Saraph, Xiaodong Wang, Brandon Reagen, Gu-Yeon Wei, Hsien-Hsin S. Lee, Carole-Jean Wu, and David Brooks. 2020. “DeepRecSys: A System for Optimizing End-To-End At-Scale Neural Recommendation Inference”. In . The 47th IEEE/ACM International Symposium on Computer Architecture (ISCA 2020)
Liu Ke, Udit Gupta, Carole-Jean Wu, Benjamin Cho, Mark Hempstead, Brandon Reagen, Xuan Zhang, David Brooks, Vikas Chandra, Utku Diril, Amin Firoozshahian, Kim Hazelwood, Bill Jia, Hsien-Hsin Lee, Meng Li, Bert Maher, Dheevatsa Mudigere, Maxim Naumov, Martin Schatz, Mikhail Smelyanskiy, and Xiaodong Wang. 2020. “RecNMP: Accelerating Personalized Recommendation With Near-Memory Processing”. In . The 47th IEEE/ACM International Symposium on Computer Architecture (ISCA 2020)
Liu Ke, Udit Gupta, Carole-Jean Wu, Benjamin Cho, Mark Hempstead, Brandon Reagen, Xuan Zhang, David Brooks, Vikas Chandra, Utku Diril, Amin Firoozshahian, Kim Hazelwood, Bill Jia, Hsien-Hsin Lee, Meng Li, Bert Maher, Dheevatsa Mudigere, Maxim Naumov, Martin Schatz, Mikhail Smelyanskiy, and Xiaodong Wang. 2020. “RecNMP: Accelerating Personalized Recommendation With Near-Memory Processing”. In . The 47th IEEE/ACM International Symposium on Computer Architecture (ISCA 2020)
Udit Gupta, Carole Wu, Xiaodong Wang, Maxim Naumov, Brandon Reagen, David Brooks, Bradford Cottel, Kim Hazelwood, Bill Jia, Hsien-Hsin Lee, Andrey Malevich, Dheevatsa Mudigere, Mikhail Smelyanskiy, Liang Xiong, and Xuan Zhang. 2020. “The Architectural Implications of Facebook’s DNN-Based Personalized Recommendation”. In . The 26th IEEE International Symposium on High-Performance Computer Architecture
Udit Gupta, Carole Wu, Xiaodong Wang, Maxim Naumov, Brandon Reagen, David Brooks, Bradford Cottel, Kim Hazelwood, Bill Jia, Hsien-Hsin Lee, Andrey Malevich, Dheevatsa Mudigere, Mikhail Smelyanskiy, Liang Xiong, and Xuan Zhang. 2020. “The Architectural Implications of Facebook’s DNN-Based Personalized Recommendation”. In . The 26th IEEE International Symposium on High-Performance Computer Architecture

2019

Paul Whatmough, Sae Lee, Marco Donato, Hsea Hsueh, Sam Xi, Udit Gupta, Lillian Pentecost, Glenn Ko, David Brooks, and Gu Wei. 2019. “A 16nm 25mm2 SoC With a 54.5x Flexibility-Efficiency Range from Dual-Core Arm Cortex-A53 to EFPGA and Cache-Coherent Accelerators”. Symposium on VLSI Circuits
Paul Whatmough, Sae Lee, Marco Donato, Hsea Hsueh, Sam Xi, Udit Gupta, Lillian Pentecost, Glenn Ko, David Brooks, and Gu Wei. 2019. “A 16nm 25mm2 SoC With a 54.5x Flexibility-Efficiency Range from Dual-Core Arm Cortex-A53 to EFPGA and Cache-Coherent Accelerators”. Symposium on VLSI Circuits