Junjie Yang

Provably Faster Algorithms for Bilevel Optimization

2021 · arXiv (Cornell University)

Bilevel optimization has been widely applied in many important machine learning applications such as hyperparameter optimization and meta-learning. Recently, several momentum-based algorithms have been proposed to solve bilevel optimization problems faster. However, those momentum-based algorithms …

RankGraph: Unified Heterogeneous Graph Learning for Cross-Domain Recommendation

2025 · arXiv (Cornell University)

Cross-domain recommendation systems face the challenge of integrating fine-grained user and item relationships across various product domains. To address this, we introduce RankGraph, a scalable graph learning framework designed to serve as a core component …

Retrospective Reader for Machine Reading Comprehension

2021 · Proceedings of the AAAI Conference on Artificial Intelligence

Machine reading comprehension (MRC) is an AI challenge that requires machines to determine the correct answers to questions based on a given passage. MRC systems must not only answer questions when necessary but also tactfully …