Publications

Jason Ansel, Edward Yang, Horace He, Natalia Gimelshein, Animesh Jain, Michael Voznesensky, Bin Bao, Peter Bell, David Berard, Evgeni Burovski, Geeta Chauhan, Anjali Chourdia, Will Constable, Alban Desmaison, Zachary DeVito, Elias Ellison, Will Feng, Jiong Gong, Michael Gschwind, Brian Hirsh, Sherlock Huang, Kshiteej Kalambarkar, Laurent Kirsch, Michael Lazos, Mario Lezcano, Yanbo Liang, Jason Liang, Yinghai Lu, C. K. Luk, Bert Maher, Yunjie Pan, Christian Puhrsch, Matthias Reso, Mark Saroufim, Marcos Yukio Siraichi, Helen Suk, Shunting Zhang, Michael Suo, Phil Tillet, Xu Zhao, Eikan Wang, Keren Zhou, Richard Zou, Xiaodong Wang, Ajit Mathews, William Wen, Gregory Chanan, Peng Wu, Soumith Chintala (2024). PyTorch 2: Faster Machine Learning Through Dynamic Python Bytecode Transformation and Graph Compilation. Proceedings of the 29th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS).

Cite Project DOI URL

Keren Zhou, Karthik Ganapathi Subramanian, Po-Hsun Lin, Matthias Fey, Binqian Yin, Jiajia Li (2024). FASTEN: Fast GPU-accelerated Segmented Matrix Multiplication for Heterogenous Graph Neural Networks. Proceedings of the 38th ACM International Conference on Supercomputing (ICS).

Cite Project DOI URL

Zhen Xie, Murali Emani, Xiaodong Yu, Dingwen Tao, Xin He, Pengfei Su, Keren Zhou, Venkatram Vishwanath (2024). Centimani: Enabling Fast AI Accelerator Selection for DNN Training with a Novel Performance Predictor. USENIX Annual Technical Conference (USENIX ATC).

Cite Project URL

Aditya Desai, Keren Zhou, Anshumali Shrivastava (2023). Hardware-Aware Compression with Random Operation Access Specific Tile (ROAST) Hashing. Proceedings of the 40th International Conference on Machine Learning (ICML).

Cite Project URL

Binqian Yin, Qinhong Hu, Yingying Zhu, Keren Zhou (2023). Semi-supervised learning for shale image segmentation with fast normalized cut loss. Geoenergy Science and Engineering.

Mao Lin, Keren Zhou, Pengfei Su (2023). DrGPUM: Guiding Memory Optimization for GPU-Accelerated Applications. Proceedings of the 28th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS).

Cite Project DOI URL

Keren Zhou, Yueming Hao, John Mellor-Crummey, Xiaozhu Meng, Xu Liu (2022). ValueExpert: Exploring Value Patterns in GPU-Accelerated Applications. Proceedings of the 27th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS).

Cite Project DOI URL

Binqian Yin, Qinhong Hu, Yingying Zhu, Chen Zhao, Keren Zhou (2022). Paw-Net: Stacking Ensemble Deep Learning for Segmenting Scanning Electron Microscopy Images of Fine-grained Shale Samples. Computers & Geosciences.

Keren Zhou, Jonathon Anderson, Xiaozhu Meng, John Mellor-Crummey (2022). Low Overhead and Context Sensitive Profiling of GPU-Accelerated Applications. Proceedings of the 36th ACM International Conference on Supercomputing (ICS).

Cite Project DOI URL

Keren Zhou, Xiaozhu Meng, Ryuichi Sai, Dejan Grubisic, John Mellor-Crummey (2022). An Automated Tool for Analysis and Tuning of GPU-Accelerated Code in HPC Applications. IEEE Transactions on Parallel and Distributed Systems (TPDS).

Cite Project DOI URL

Ryuichi Sai, John Mellor-Crummey, Xiaozhu Meng, Keren Zhou, Mauricio Araya-Polo, Jie Meng (2022). Accelerating High-order Stencils on GPUs. Concurrency and Computation: Practice and Experience (CCPE).

Cite Project DOI URL

Keren Zhou, Xiaozhu Meng, Ryuichi Sai, John Mellor-Crummey (2021). GPA: A GPU Performance Advisor Based on Instruction Sampling. IEEE/ACM International Symposium on Code Generation and Optimization (CGO).

Cite Project DOI URL

Aaron Thomas Cherian, Keren Zhou, Dejan Grubisic, Xiaozhu Meng, John Mellor-Crummey (2021). Measurement and Analysis of GPU-Accelerated OpenCL Computations on Intel GPUs. IEEE/ACM International Workshop on Programming and Performance Visualization Tools (ProTools).

Cite Project DOI URL

Keren Zhou, Laksono Adhianto, Jonathon Anderson, Aaron Cherian, Dejan Grubisic, Mark Krentel, Yumeng Liu, Xiaozhu Meng, John Mellor-Crummey (2021). Measurement and Analysis of GPU-accelerated Applications with HPCToolkit. Parallel Computing (PARCO).

Cite Project DOI URL

Keren Zhou, Mark W. Krentel, John Mellor-Crummey (2020). Tools for Top-down Performance Analysis of GPU-Accelerated Applications. Proceedings of the 34th ACM International Conference on Supercomputing (ICS).

Cite Project DOI URL

Keren Zhou, Yueming Hao, John Mellor-Crummey, Xiaozhu Meng, Xu Liu (2020). GVPROF: A Value Profiler for GPU-Based Clusters. International Conference for High Performance Computing, Networking, Storage and Analysis (SC).

Cite Project DOI URL

Keren Zhou, Mark Krentel, John Mellor-Crummey (2020). A Tool for Top-down Performance Analysis of GPU-Accelerated Applications. Proceedings of the 25th ACM SIGPLAN Symposium on Principles and Practice of Parallel Programming (PPoPP).

Cite Project DOI URL

Keren Zhou, John Mellor-Crummey (2019). A Tool for Performance Analysis of GPU-Accelerated Applications. IEEE/ACM International Symposium on Code Generation and Optimization (CGO).

Cite Project DOI URL

Keren Zhou, Guangming Tan, Wei Zhou (2018). Quadboost: A Scalable Concurrent Quadtree. IEEE Transactions on Parallel and Distributed Systems (TPDS).

Xiuxia Zhang, Guangming Tan, Shuangbai Xue, Jiajia Li, Keren Zhou, Mingyu Chen (2017). Understanding the GPU Microarchitecture to Achieve Bare-Metal Performance Tuning. Proceedings of the 22nd ACM SIGPLAN Symposium on Principles and Practice of Parallel Programming (PPoPP).

Cite Project DOI URL

Keren Zhou, Guangming Tan, Xiuxia Zhang, Chaowei Wang, Ninghui Sun (2017). A Performance Analysis Framework for Exploiting GPU Microarchitectural Capability. Proceedings of the International Conference on Supercomputing (ICS).

Cite Project DOI URL

Qiang Li, Maojie Gu, Keren Zhou, Xiaoming Sun (2015). Multi-Classes Feature Engineering with Sliding Window for Purchase Prediction in Mobile Commerce. IEEE International Conference on Data Mining Workshop (ICDMW).

Zilong Tan, Keren Zhou, Hao Zhang, Wei Zhou (2015). BF-MapReduce: A Bloom Filter Based Efficient Lightweight Search. IEEE Conference on Collaboration and Internet Computing (CIC).