← Search

Yiwen Kou

10 accepted papers

2025

On the Power of Multitask Representation Learning with Gradient Descent

AISTATS 2025poster

Representation learning, particularly multi-task representation learning, has gained widespread popularity in various deep learning applications, ranging from computer vision to natural language processing, due to its remarkable generalization performance. Despite its growing use, our understanding…

Cited by 0SourceScholar
2024

Fast Sampling via Discrete Non-Markov Diffusion Models with Predetermined Transition Time

NeurIPS 2024poster

Discrete diffusion models have emerged as powerful tools for high-quality data generation. Despite their success in discrete spaces, such as text generation tasks, the acceleration of discrete diffusion models remains under-explored. In this paper, we propose discrete non-Markov diffusion models (DN…

2024

GraphVis: Boosting LLMs with Visual Knowledge Graph Integration

NeurIPS 2024poster

The rapid evolution of large language models (LLMs) has expanded their capabilities across various data modalities, extending from well-established image data to increasingly popular graph data. Given the limitation of LLMs in hallucinations and inaccuracies in recalling factual knowledge, Knowledge…

Cited by 3SourcePDFScholar
2024

Matching the Statistical Query Lower Bound for $k$-Sparse Parity Problems with Sign Stochastic Gradient Descent

NeurIPS 2024poster

The $k$-sparse parity problem is a classical problem in computational complexity and algorithmic theory, serving as a key benchmark for understanding computational classes. In this paper, we solve the $k$-sparse parity problem with sign stochastic gradient descent, a variant of stochastic gradient d…

Cited by 0SourcePDFScholar
2023

Benign Overfitting in Two-layer ReLU Convolutional Neural Networks

ICML 2023poster

Modern deep learning models with great expressive power can be trained to overfit the training data but still generalize well. This phenomenon is referred to as benign overfitting. Recently, a few studies have attempted to theoretically understand benign overfitting in neural networks. However, thes…

Cited by 49SourcePDFScholar
2023

How Does Semi-supervised Learning with Pseudo-labelers Work? A Case Study

ICLR 2023poster

Semi-supervised learning is a popular machine learning paradigm that utilizes a large amount of unlabeled data as well as a small amount of labeled data to facilitate learning tasks. While semi-supervised learning has achieved great success in training neural networks, its theoretical understanding…

Cited by 9SourcePDFScholar
2023

Implicit Bias of Gradient Descent for Two-layer ReLU and Leaky ReLU Networks on Nearly-orthogonal Data

NeurIPS 2023poster

The implicit bias towards solutions with favorable properties is believed to be a key reason why neural networks trained by gradient-based optimization can generalize well. While the implicit bias of gradient flow has been widely studied for homogeneous neural networks (including ReLU and leaky ReLU…

Cited by 16SourcePDFScholar
2023

Why Does Sharpness-Aware Minimization Generalize Better Than SGD?

NeurIPS 2023poster

The challenge of overfitting, in which the model memorizes the training data and fails to generalize to test data, has become increasingly significant in the training of large neural networks. To tackle this challenge, Sharpness-Aware Minimization (SAM) has emerged as a promising training method, wh…

Cited by 19SourcePDFScholar