← Search

Jiarui Feng

8 accepted papers

2026

DAG-MoE: From Simple Mixture to Structural Aggregation in Mixture-of-Experts

ICML 2026poster

Mixture-of-Experts (MoE) models have become a leading approach for decoupling parameter count from computational cost in large language models. Despite significant progress, effectively scaling MoE performance remains a challenge. Previous work shows that the use of fine-grained experts enlarges the…

Cited by 0SourceScholar
2025

GOFA: A Generative One-For-All Model for Joint Graph Language Modeling

ICLR 2025poster

Foundation models, such as Large Language Models (LLMs) or Large Vision Models (LVMs), have emerged as one of the most powerful tools in the respective fields. However, unlike text and image data, graph data do not have a definitive structure, posing great challenges to developing a Graph Foundation…

2024

Can Graph Learning Improve Planning in LLM-based Agents?

NeurIPS 2024poster

Task planning in language agents is emerging as an important research topic alongside the development of large language models (LLMs). It aims to break down complex user requests in natural language into solvable sub-tasks, thereby fulfilling the original requests. In this context, the sub-tasks can…

2024

One For All: Towards Training One Graph Model For All Classification Tasks

ICLR 2024spotlight

Designing a single model to address multiple tasks has been a long-standing objective in artificial intelligence. Recently, large language models have demonstrated exceptional capability in solving different tasks within the language domain. However, a unified model for various graph tasks remains u…

2023

Distance-Restricted Folklore Weisfeiler-Leman GNNs with Provable Cycle Counting Power

NeurIPS 2023spotlight

The ability of graph neural networks (GNNs) to count certain graph substructures, especially cycles, is important for the success of GNNs on a wide range of tasks. It has been recently used as a popular metric for evaluating the expressive power of GNNs. Many of the proposed GNN models with provable…

2023

Extending the Design Space of Graph Neural Networks by Rethinking Folklore Weisfeiler-Lehman

NeurIPS 2023poster

Message passing neural networks (MPNNs) have emerged as the most popular framework of graph neural networks (GNNs) in recent years. However, their expressive power is limited by the 1-dimensional Weisfeiler-Lehman (1-WL) test. Some works are inspired by $k$-WL/FWL (Folklore WL) and design the corres…

2023

MAG-GNN: Reinforcement Learning Boosted Graph Neural Network

NeurIPS 2023poster

While Graph Neural Networks (GNNs) recently became powerful tools in graph learning tasks, considerable efforts have been spent on improving GNNs' structural encoding ability. A particular line of work proposed subgraph GNNs that use subgraph information to improve GNNs' expressivity and achieved gr…

Cited by 20SourcePDFScholar
2022

How Powerful are K-hop Message Passing Graph Neural Networks

NeurIPS 2022accept

The most popular design paradigm for Graph Neural Networks (GNNs) is 1-hop message passing---aggregating information from 1-hop neighbors repeatedly. However, the expressive power of 1-hop message passing is bounded by the Weisfeiler-Lehman (1-WL) test. Recently, researchers extended 1-hop message p…