Search Results for author: Yifei Shen

Found 27 papers, 13 papers with code

Benchmarking Graphormer on Large-Scale Molecular Modeling Datasets

3 code implementations9 Mar 2022 Yu Shi, Shuxin Zheng, Guolin Ke, Yifei Shen, Jiacheng You, Jiyan He, Shengjie Luo, Chang Liu, Di He, Tie-Yan Liu

This technical note describes the recent updates of Graphormer, including architecture design modifications, and the adaption to 3D molecular dynamics simulation.

Benchmarking Graph Regression +1

Sparse Mixture-of-Experts are Domain Generalizable Learners

1 code implementation8 Jun 2022 Bo Li, Yifei Shen, Jingkang Yang, Yezhen Wang, Jiawei Ren, Tong Che, Jun Zhang, Ziwei Liu

It is motivated by an empirical finding that transformer-based models trained with empirical risk minimization (ERM) outperform CNN-based models employing state-of-the-art (SOTA) DG algorithms on multiple DG datasets.

Ranked #10 on Domain Generalization on DomainNet (using extra training data)

Domain Generalization Object Recognition

Graph Neural Networks for Wireless Communications: From Theory to Practice

1 code implementation21 Mar 2022 Yifei Shen, Jun Zhang, S. H. Song, Khaled B. Letaief

For design guidelines, we propose a unified framework that is applicable to general design problems in wireless networks, which includes graph modeling, neural architecture design, and theory-guided performance enhancement.

Graph Neural Networks for Scalable Radio Resource Management: Architecture Design and Theoretical Analysis

1 code implementation15 Jul 2020 Yifei Shen, Yuanming Shi, Jun Zhang, Khaled B. Letaief

In this paper, we propose to apply graph neural networks (GNNs) to solve large-scale radio resource management problems, supported by effective neural network architecture design and theoretical analysis.

Computational Efficiency Distributed Optimization +1

SIMPLE: Specialized Model-Sample Matching for Domain Generalization

1 code implementation International Conference on Learning Representations 2023 Ziyue Li, Kan Ren, Xinyang Jiang, Yifei Shen, Haipeng Zhang, Dongsheng Li

Moreover, our method is highly efficient and achieves more than 1000 times training speedup compared to the conventional DG methods with fine-tuning a pretrained model.

Domain Generalization

How Powerful is Graph Convolution for Recommendation?

1 code implementation17 Aug 2021 Yifei Shen, Yongji Wu, Yao Zhang, Caihua Shan, Jun Zhang, Khaled B. Letaief, Dongsheng Li

In this paper, we endeavor to obtain a better understanding of GCN-based CF methods via the lens of graph signal processing.

Collaborative Filtering

Hybrid Far- and Near-Field Channel Estimation for THz Ultra-Massive MIMO via Fixed Point Networks

1 code implementation10 May 2022 Wentao Yu, Yifei Shen, Hengtao He, Xianghao Yu, Jun Zhang, Khaled B. Letaief

We draw inspirations from fixed point theory to develop an efficient deep learning based channel estimator with adaptive complexity and linear convergence guarantee.

An Adaptive and Robust Deep Learning Framework for THz Ultra-Massive MIMO Channel Estimation

1 code implementation29 Nov 2022 Wentao Yu, Yifei Shen, Hengtao He, Xianghao Yu, Shenghui Song, Jun Zhang, Khaled B. Letaief

For practical usage, the proposed framework is further extended to wideband THz UM-MIMO systems with beam squint effect.

A Graph Neural Network Approach for Scalable Wireless Power Control

2 code implementations19 Jul 2019 Yifei Shen, Yuanming Shi, Jun Zhang, Khaled B. Letaief

Specifically, a $K$-user interference channel is first modeled as a complete graph, where the quantitative information of wireless channels is incorporated as the features of the graph.

Complete Dictionary Learning via $\ell_p$-norm Maximization

1 code implementation24 Feb 2020 Yifei Shen, Ye Xue, Jun Zhang, Khaled B. Letaief, Vincent Lau

Dictionary learning is a classic representation learning method that has been widely applied in signal processing and data analytics.

Computational Efficiency Dictionary Learning +1

Decentralized Statistical Inference with Unrolled Graph Neural Networks

1 code implementation4 Apr 2021 He Wang, Yifei Shen, Ziyuan Wang, Dongsheng Li, Jun Zhang, Khaled B. Letaief, Jie Lu

In this paper, we investigate the decentralized statistical inference problem, where a network of agents cooperatively recover a (structured) vector from private noisy samples without centralized coordination.

Learning Domain Invariant Prompt for Vision-Language Models

1 code implementation8 Dec 2022 Cairong Zhao, Yubin Wang, Xinyang Jiang, Yifei Shen, Kaitao Song, Dongsheng Li, Duoqian Miao

Prompt learning is one of the most effective and trending ways to adapt powerful vision-language foundation models like CLIP to downstream datasets by tuning learnable prompt vectors with very few samples.

Domain Generalization Language Modelling +2

LORM: Learning to Optimize for Resource Management in Wireless Networks with Few Training Samples

no code implementations18 Dec 2018 Yifei Shen, Yuanming Shi, Jun Zhang, Khaled B. Letaief

To further address the task mismatch problem, we develop a transfer learning method via self-imitation in LORM, named LORM-TL, which can quickly adapt a pre-trained machine learning model to the new task with only a few additional unlabeled training samples.

BIG-bench Machine Learning Imitation Learning +2

Transfer Learning for Mixed-Integer Resource Allocation Problems in Wireless Networks

no code implementations17 Nov 2018 Yifei Shen, Yuanming Shi, Jun Zhang, Khaled B. Letaief

A unique advantage of the proposed method is that it can tackle the task mismatch issue with a few additional unlabeled training samples, which is especially important when transferring to large-size problems.

Transfer Learning

Blind Data Detection in Massive MIMO via $\ell_3$-norm Maximization over the Stiefel Manifold

no code implementations26 Apr 2020 Ye Xue, Yifei Shen, Vincent Lau, Jun Zhang, Khaled B. Letaief

Specifically, we propose a novel $\ell_3$-norm-based formulation to recover the data without channel estimation.

Invariant Information Bottleneck for Domain Generalization

no code implementations11 Jun 2021 Bo Li, Yifei Shen, Yezhen Wang, Wenzhen Zhu, Colorado J. Reed, Jun Zhang, Dongsheng Li, Kurt Keutzer, Han Zhao

IIB significantly outperforms IRM on synthetic datasets, where the pseudo-invariant features and geometric skews occur, showing the effectiveness of proposed formulation in overcoming failure modes of IRM.

Domain Generalization

Neural Calibration for Scalable Beamforming in FDD Massive MIMO with Implicit Channel Estimation

no code implementations3 Aug 2021 Yifan Ma, Yifei Shen, Xianghao Yu, Jun Zhang, S. H. Song, Khaled B. Letaief

Channel estimation and beamforming play critical roles in frequency-division duplexing (FDD) massive multiple-input multiple-output (MIMO) systems.

Learn to Communicate with Neural Calibration: Scalability and Generalization

no code implementations1 Oct 2021 Yifan Ma, Yifei Shen, Xianghao Yu, Jun Zhang, S. H. Song, Khaled B. Letaief

Furthermore, such networks will vary dynamically in a significant way, which makes it intractable to develop comprehensive analytical models.

Computational Efficiency Management

Reinforcement Learning Enhanced Explainer for Graph Neural Networks

no code implementations NeurIPS 2021 Caihua Shan, Yifei Shen, Yao Zhang, Xiang Li, Dongsheng Li

To address these issues, we propose a RL-enhanced GNN explainer, RG-Explainer, which consists of three main components: starting point selection, iterative graph generation and stopping criteria learning.

Combinatorial Optimization Graph Generation +2

Neural Piecewise-Constant Delay Differential Equations

no code implementations4 Jan 2022 Qunxi Zhu, Yifei Shen, Dongsheng Li, Wei Lin

Continuous-depth neural networks, such as the Neural Ordinary Differential Equations (ODEs), have aroused a great deal of interest from the communities of machine learning and data science in recent years, which bridge the connection between deep neural networks and dynamical systems.

An Empirical Study of Graphormer on Large-Scale Molecular Modeling Datasets

no code implementations28 Feb 2022 Yu Shi, Shuxin Zheng, Guolin Ke, Yifei Shen, Jiacheng You, Jiyan He, Shengjie Luo, Chang Liu, Di He, Tie-Yan Liu

This technical note describes the recent updates of Graphormer, including architecture design modifications, and the adaption to 3D molecular dynamics simulation.

Dissecting Arbitrary-scale Super-resolution Capability from Pre-trained Diffusion Generative Models

no code implementations1 Jun 2023 Ruibin Li, Qihua Zhou, Song Guo, Jie Zhang, Jingcai Guo, Xinyang Jiang, Yifei Shen, Zhenhua Han

Diffusion-based Generative Models (DGMs) have achieved unparalleled performance in synthesizing high-quality visual content, opening up the opportunity to improve image super-resolution (SR) tasks.

Image Super-Resolution

Large Language Models Empowered Autonomous Edge AI for Connected Intelligence

no code implementations6 Jul 2023 Yifei Shen, Jiawei Shao, Xinjie Zhang, Zehong Lin, Hao Pan, Dongsheng Li, Jun Zhang, Khaled B. Letaief

The evolution of wireless networks gravitates towards connected intelligence, a concept that envisions seamless interconnectivity among humans, objects, and intelligence in a hyper-connected cyber-physical world.

Code Generation Federated Learning +3

Label Propagation for Graph Label Noise

no code implementations25 Oct 2023 Yao Cheng, Caihua Shan, Yifei Shen, Xiang Li, Siqiang Luo, Dongsheng Li

In this paper, we study graph label noise in the context of arbitrary heterophily, with the aim of rectifying noisy labels and assigning labels to previously unlabeled nodes.

Denoising Node Classification

Toward Open-ended Embodied Tasks Solving

no code implementations10 Dec 2023 William Wei Wang, Dongqi Han, Xufang Luo, Yifei Shen, Charles Ling, Boyu Wang, Dongsheng Li

Empowering embodied agents, such as robots, with Artificial Intelligence (AI) has become increasingly important in recent years.

Understanding Training-free Diffusion Guidance: Mechanisms and Limitations

no code implementations19 Mar 2024 Yifei Shen, Xinyang Jiang, Yezhen Wang, Yifan Yang, Dongqi Han, Dongsheng Li

Adding additional control to pretrained diffusion models has become an increasingly popular research area, with extensive applications in computer vision, reinforcement learning, and AI for science.

Cannot find the paper you are looking for? You can Submit a new open access paper.