Yueqi Wang

Yueqi Wang

Machine Learning Engineer

Hi, I’m a Machine Learning Engineer at Google, where I work on YouTube’s recommender systems. My work focuses on LLM-based generative recommendation, large-scale retrieval models, multimodal content understanding, user cold-start, etc.

Before Google, I obtained a PhD in Neuroscience from the University of Utah, where I studied human brain development using neurophysiology and computational genomic technologies, advised by Dr. Alex Shcheglovitov.

In addition, I received a MS in Computer Science from Columbia University, and a BS in Biological Sciences from Peking University.

Selected Publications

1. Machine Learning

  • Vectorizing the Trie: Efficient Constrained Decoding for LLM-based Generative Retrieval on Accelerators
    Z Su*, I Katsman*, Y Wang*, R He, L Heldt, R Keshavan, SC Wang, X Yi, M Gao, O Dalal, L Hong, E Chi, N Han
    arXiv preprint, 2026
    Paper GitHub
  • PLUM: Adapting Pre-trained Language Models for Industrial-scale Generative Recommendations
    R He*, L Heldt*, L Hong*, R Keshavan*, S Mao*, N Mehta*, Z Su*, A Tsai*, Y Wang*, SC Wang*, X Yi*, L Baugher, B Cakici, E Chi, C Goodrow, N Han, H Ma, R Rosales, A Van Soest, D Tandon, SL Wu, W Yang, Y Zheng
    WWW, 2025
    Paper
  • LLM-Powered Nuanced Video Attribute Annotation for Enhanced Recommendations
    B Long*, Y Wang*, H Mehta, M Zomnir, O Pathak, C Meng, R Jia, Y Peng, D Hong, X Wu, M Gao, O Dalal, N Han
    RecSys, 2025
    Paper
  • Amortized Probabilistic Detection of Communities in Graphs
    Y Wang*, Y Lee*, P Basu, J Lee, YW Teh, L Paninski, A Pakman
    ICML SPIGM Workshop, 2024
    Paper GitHub
  • LLMs for user interest exploration in large-scale recommendation systems
    J Wang, H Lu, Y Liu, H Ma, Y Wang, Y Gu, S Zhang, N Han, S Bi, L Baugher, E Chi, M Chen
    RecSys, 2024
    Paper
  • Learning from negative user feedback and measuring responsiveness for sequential recommenders
    Y Wang, Y Halpern, S Chang, J Feng, EY Le, L Li, X Liang, MC Huang, S Li, A Beutel, Y Zhang, S Bi
    RecSys, 2023
    Paper
  • Neural Clustering Processes
    A Pakman, Y Wang, C Mitelut, J Lee, L Paninski
    ICML, 2020
    Paper GitHub
  • Spike Sorting using the Neural Clustering Process
    Y Wang, A Pakman, C Mitelut, J Lee, L Paninski
    NeurIPS NeuroAI Workshop, 2019
    Paper GitHub

2. Neuroscience

  • Modeling human telencephalic development and autism-associated SHANK3 deficiency using organoids generated from single neural rosettes
    Y Wang, S Chiola, G Yang, C Russell, CJ Armstrong, Y Wu, J Spampanato, P Tarboton, HMA Ullah, NU Edgar, AN Chang, DA Harmin, V Dickinson Bocchi, E Vezzoli, D Besusso, J Cui, E Cattaneo, J Kubanek, A Shcheglovitov
    Nature communications, 2022
    Paper GitHub Data Viewer
  • Defective AMPA-mediated synaptic transmission and morphology in human neurons with hemizygous SHANK3 deletion engrafted in mouse prefrontal cortex
    S Chiola*, KL Napan*, Y Wang*, RM Lazarenko, CJ Armstrong, J Cui, A Shcheglovitov
    Molecular psychiatry, 2021
    Paper
  • Heterophilic type II cadherins are required for high-magnitude synaptic potentiation in the hippocampus
    R Basu, X Duan, MR Taylor, EA Martin, S Muralidhar, Y Wang, L Gangi-Wellman, SC Das, M Yamagata, PJ West, JR Sanes, ME Williams
    Neuron, 2017
    Paper

Full publication list on Google Scholar.