Demo Data from SIGIR 2026 / reviewers in the wild / expert
Junfeng Fang
dblp:340/7929
· DBLP profile ↗
43ranked-venue papers
8as first author
43since 2021 · last 2026
0000-0002-3317-2103ORCID · conflict
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 31 · 5 first-author · 31 since 2021Databases, data management, data science and information retrieval · 14 · 4 first-author · 14 since 2021Applied, interdisciplinary, general and emerging computing · 8 · 1 first-author · 8 since 2021Graphics, computer vision, multimedia, augmented reality and games · 6 · 2 first-author · 6 since 2021
Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.
| Artificial intelligence
37 papers |
Trustworthy machine learning · 25% Language models and text generation · 24% Graph learning · 20% | |
| Databases, data mining, and information retrieval
4 papers |
Data mining · 67% Recommender systems · 18% Information retrieval · 16% |
Topics — the 30 heaviest of 90, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Machine learning › Graph learning
graph neural network |
6.7 | 11 | 2026 | G-Designer: Architecting Multi-agent Communication Topologies via Graph Neural Networks · ICML 2025 Invariant Graph Learning for Causal Effect Estimation · WWW 2024 Graph Anomaly Detection with Bi-level Optimization · WWW 2024 |
Natural language and speech › Language models and text generation
knowledge editing |
6.5 | 8 | 2026 | LPEdit: Locality-Preserving Knowledge Editing for MultiModal Large Language Models · WWW 2026 Explainable and Efficient Editing for Large Language Models · WWW 2025 AnyEdit: Edit Any Knowledge Encoded in Language Models · ICML 2025 |
Machine learning › Trustworthy machine learning
interpretability |
5.1 | 7 | 2026 | Out-of-Distribution-Resistant Evaluations for Explanations of Graph Neural Networks · IEEE Trans. Pattern Anal. Mach. Intell. 2026 SEHG: Bridging Interpretability and Prediction in Self-Explainable Heterogeneous Graph Neural Networks · WWW 2025 On the Role of Attention Heads in Large Language Model Safety · ICLR 2025 |
Machine learning › Trustworthy machine learning › interpretability
graph neural network explanation |
3.1 | 4 | 2026 | Out-of-Distribution-Resistant Evaluations for Explanations of Graph Neural Networks · IEEE Trans. Pattern Anal. Mach. Intell. 2026 EXGC: Bridging Efficiency and Explainability in Graph Condensation · WWW 2024 Cooperative Explanations of Graph Neural Networks · WSDM 2023 |
Computer vision › Vision and language › vision-language model
multimodal large language model |
2.9 | 4 | 2026 | LPEdit: Locality-Preserving Knowledge Editing for MultiModal Large Language Models · WWW 2026 DAMA: Data- and Model-aware Alignment of Multi-modal LLMs · ICML 2025 Towards Neuron Attributions in Multi-Modal Large Language Models · NeurIPS 2024 |
Natural language and speech › Language models and text generation
hallucination mitigation |
2.7 | 3 | 2026 | Mitigating Hallucinations in Large Vision-Language Models without Performance Degradation · ACL (1) 2026 DAMA: Data- and Model-aware Alignment of Multi-modal LLMs · ICML 2025 AlphaEdit: Null-Space Constrained Knowledge Editing for Language Models · ICLR 2025 |
Machine learning › Trustworthy machine learning
robustness |
2.7 | 3 | 2026 | Out-of-Distribution-Resistant Evaluations for Explanations of Graph Neural Networks · IEEE Trans. Pattern Anal. Mach. Intell. 2026 RMBRec: Robust Multi-Behavior Recommendation towards Target Behaviors · WWW 2026 Evaluating Post-hoc Explanations for Graph Neural Networks via Robustness Analysis · NeurIPS 2023 |
Machine learning › Trustworthy machine learning › hallucination
vision-language model hallucination |
1.9 | 2 | 2026 | Mitigating Hallucinations in Large Vision-Language Models without Performance Degradation · ACL (1) 2026 Cracking the Code of Hallucination in LVLMs with Vision-aware Head Divergence · ACL (1) 2025 |
Machine learning › Generative modeling
diffusion model |
1.7 | 2 | 2025 | ACE: Concept Editing in Diffusion Models without Performance Degradation · ACM Multimedia 2025 NExT-Mol: 3D Diffusion Meets 1D Language Modeling for 3D Molecule Generation · ICLR 2025 |
Knowledge, reasoning and agents › Multi-agent systems
LLM-based multi-agent systems |
1.7 | 2 | 2025 | G-Designer: Architecting Multi-agent Communication Topologies via Graph Neural Networks · ICML 2025 Multi-agent Architecture Search via Agentic Supernet · ICML 2025 |
Machine learning › Trustworthy machine learning › robustness
adversarial robustness |
1.7 | 2 | 2026 | Out-of-Distribution-Resistant Evaluations for Explanations of Graph Neural Networks · IEEE Trans. Pattern Anal. Mach. Intell. 2026 Evaluating Post-hoc Explanations for Graph Neural Networks via Robustness Analysis · NeurIPS 2023 |
Machine learning › Trustworthy machine learning › robustness › distributional robustness
invariant risk minimization |
1.0 | 1 | 2026 | RMBRec: Robust Multi-Behavior Recommendation towards Target Behaviors · WWW 2026 |
Natural language and speech › Language models and text generation › knowledge editing
multimodal knowledge editing |
1.0 | 1 | 2026 | LPEdit: Locality-Preserving Knowledge Editing for MultiModal Large Language Models · WWW 2026 |
Machine learning › Representation and self-supervised learning
multimodal representation learning |
1.0 | 1 | 2026 | Hierarchical Semantic Alignment for Image Clustering · AAAI 2026 |
Machine learning › Representation and self-supervised learning
semantic alignment |
1.0 | 1 | 2026 | Hierarchical Semantic Alignment for Image Clustering · AAAI 2026 |
Data mining
clustering |
1.0 | 1 | 2026 | Hierarchical Semantic Alignment for Image Clustering · AAAI 2026 |
Data mining › clustering
image clustering |
1.0 | 1 | 2026 | Hierarchical Semantic Alignment for Image Clustering · AAAI 2026 |
Recommender systems › implicit feedback learning
multi-behavior recommendation |
1.0 | 1 | 2026 | RMBRec: Robust Multi-Behavior Recommendation towards Target Behaviors · WWW 2026 |
Machine learning › Generative modeling › molecular generation
3d molecule generation |
0.9 | 1 | 2025 | NExT-Mol: 3D Diffusion Meets 1D Language Modeling for 3D Molecule Generation · ICLR 2025 |
Machine learning › Reinforcement learning › multi-agent reinforcement learning › multi-agent communication
communication topology design |
0.9 | 1 | 2025 | G-Designer: Architecting Multi-agent Communication Topologies via Graph Neural Networks · ICML 2025 |
Machine learning › Generative modeling › diffusion model › controllable generation
concept editing |
0.9 | 1 | 2025 | ACE: Concept Editing in Diffusion Models without Performance Degradation · ACM Multimedia 2025 |
Natural language and speech › Language models and text generation › hallucination detection
context faithfulness |
0.9 | 1 | 2025 | Is Factuality Enhancement a Free Lunch For LLMs? Better Factuality Can Lead to Worse Context-Faithfulness · ICLR 2025 |
Natural language and speech › Language models and text generation › trustworthy language model › large language model reliability
factuality |
0.9 | 1 | 2025 | Is Factuality Enhancement a Free Lunch For LLMs? Better Factuality Can Lead to Worse Context-Faithfulness · ICLR 2025 |
Machine learning › Graph learning › graph regularization
graph information bottleneck |
0.9 | 1 | 2025 | Iterative Substructure Extraction for Molecular Relational Learning with Interactive Graph Information Bottleneck · ICLR 2025 |
Machine learning › Trustworthy machine learning
hallucination |
0.9 | 1 | 2025 | Is Factuality Enhancement a Free Lunch For LLMs? Better Factuality Can Lead to Worse Context-Faithfulness · ICLR 2025 |
Machine learning › Graph learning › graph neural network
heterogeneous graph neural network |
0.9 | 1 | 2025 | SEHG: Bridging Interpretability and Prediction in Self-Explainable Heterogeneous Graph Neural Networks · WWW 2025 |
Machine learning › Efficient and distributed learning › inference efficiency
inference cost optimization |
0.9 | 1 | 2025 | Multi-agent Architecture Search via Agentic Supernet · ICML 2025 |
Natural language and speech › Language models and text generation
large language model |
0.9 | 1 | 2025 | Neuron-Level Sequential Editing for Large Language Models · ACL (1) 2025 |
Natural language and speech › Language models and text generation
large language model safety |
0.9 | 1 | 2025 | On the Role of Attention Heads in Large Language Model Safety · ICLR 2025 |
Natural language and speech › Language models and text generation › knowledge editing
lifelong model editing |
0.9 | 1 | 2025 | Reinforced Lifelong Editing for Language Models · ICML 2025 |
Methods — techniques the papers use, named apart from their topics
graph neural network · 2.3residual attention · 2.0noun-level concepts · 2.0invariant risk minimization · 2.0caption-level semantics · 2.0contrastive decoding · 1.9OOD reweighting · 1.7mutual information maximization · 1.0hallucination mitigation · 1.0conditional graph diffusion model · 1.0adversarial attack · 1.0transfer learning · 0.9reinforcement learning · 0.9multimodal fusion · 0.9multi-task learning · 0.9latent space learning · 0.9language modeling · 0.9interactive substructure extraction · 0.9
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | Hierarchical Semantic Alignment for Image ClusteringabstractImage clustering is a classic problem in computer vision, which categorizes images into different groups. Recent studies utilize nouns as external semantic knowledge to improve clustering performance. However, these methods often overlook the inherent ambiguity of nouns, which can distort semantic representations and degrade clustering quality. To address this issue, we propose a hierarChical semAntic alignmEnt method for image clustering, dubbed CAE, which improves clustering performance in a training-free manner. In our approach, we incorporate two complementary types of textual semantics: caption-level descriptions, which convey fine-grained attributes of image content, and noun-level concepts, which represent high-level object categories. We first select relevant nouns from WordNet and descriptions from caption datasets to construct a semantic space aligned with image features. Then, we design a residual attention mechanism to further enhance the discriminability of this space. Finally, we combine the enhanced semantic and image features to perform clustering. Extensive experiments across 8 datasets demonstrate the effectiveness of our method, notably surpassing the state-of-the-art training-free approach with a 4.2% improvement in accuracy and a 2.9% improvement in adjusted rand index (ARI) on the ImageNet-1K dataset. Beier Zhu, Junfeng Fang, Shuo Wang 0008, Kesen Zhao, Hanwang Zhang |
AAAI | 4 |
| 2026 | Mitigating Hallucinations in Large Vision-Language Models without Performance DegradationabstractXingyu Zhu, Junfeng Fang, Shuo Wang, Beier Zhu, Zhicai Wang, Yonghui Yang, Xiangnan He. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Junfeng Fang, Shuo Wang 0008, Beier Zhu, Zhicai Wang, Yonghui Yang 0001, Xiangnan He 0001 |
ACL (1) | 2 |
| 2026 | An Energy-Efficient Multimodal Retrieval Framework for Inference on Heterogeneous Edge NodesabstractLarge-model-driven multimodal retrieval on mobile and edge devices must balance retrieval accuracy, inference latency, and energy efficiency under heterogeneous hardware constraints. This paper proposes an energy-efficiency-aware multimodal retrieval framework for heterogeneous edge environments. The framework first decomposes the end-to-end model into retrieval-aware functional submodules to preserve cross-modal embedding discriminability under low-precision execution. It then introduces a CPU–NPU–DSP collaborative scheduling mechanism to reduce latency and energy consumption, together with an energy-adaptive control strategy that dynamically adjusts execution paths and inference precision under varying energy budgets. Experiments on MS-COCO image–text retrieval show that the proposed method achieves an mAP of \(68.2\%\) and Recall@5 of \(90.0\%\), while reducing end-to-end latency to 118.5 ms and per-query energy consumption to 145.6 mJ. These results demonstrate an improved balance between retrieval effectiveness and edge-side efficiency. Junfeng Fang, Yan Zhang 0104, Zhaoxi Feng, Manzhou Li |
ICMR | 1 |
| 2026 | Privacy-Constrained Low-Bit Representation Learning for Person Image RetrievalabstractEfficient storage and fast search with high retrieval accuracy remain key challenges for person image retrieval at large image scales. Low-bit visual representations can reduce storage and computation costs, but existing methods often overlook privacy leakage from residual sensitive information in hash codes, allowing attackers to infer attributes such as identity, gender, or age. To address this problem, we propose a privacy-constrained low-bit representation learning framework for person image retrieval. The framework includes a semantic-consistent low-bit retrieval encoder that preserves ranking structures under extremely low-bit budgets, a privacy-utility decoupling encoder that separates retrieval semantics from privacy information, and a balancing strategy that controls the trade-off between utility and privacy through sample-level weighting and bit-level gating. Experiments show that our method achieves an mAP of \(85.6\%\) and Rank-1 accuracy of \(92.5\%\), while reducing the Attack Success Rate under deep inversion attacks from over \(90\%\) to \(53.2\%\). It also maintains low retrieval latency, requiring only 0.02 ms per query, demonstrating a favorable balance between retrieval utility, privacy protection, and efficiency. Junfeng Fang, Yan Zhang 0104, Manzhou Li, Xinjin Ge, Weiyuan Cui |
ICMR | 1 |
| 2026 | RMBRec: Robust Multi-Behavior Recommendation towards Target BehaviorsabstractMulti-behavior recommendation faces a critical challenge in practice: auxiliary behaviors (e.g., clicks, carts) are often noisy, weakly correlated, or semantically misaligned with the target behavior (e.g., purchase), which leads to biased preference learning and suboptimal performance. While existing methods attempt to fuse these heterogeneous signals, they inherently lack a principled mechanism to ensure robustness against such behavioral inconsistency. In this work, we propose Robust Multi-Behavior Recommendation towards Target Behaviors (RMBRec), a robust multi-behavior recommendation framework grounded in an information-theoretic robustness principle. We interpret robustness as a joint process of maximizing predictive information while minimizing its variance across heterogeneous behavioral environments. Under this perspective, the Representation Robustness Module (RRM) enhances local semantic consistency by maximizing the mutual information between users' auxiliary and target representations, whereas the Optimization Robustness Module (ORM) enforces global stability by minimizing the variance of predictive risks across behaviors, which is an efficient approximation to invariant risk minimization. This local-global collaboration bridges representation purification and optimization invariance in a theoretically coherent way. Extensive experiments on three real-world datasets demonstrate that RMBRec not only outperforms state-of-the-art methods in accuracy but also maintains remarkable stability under various noise perturbations. For reproducibility, our code is available at https://github.com/miaomiao-cai2/RMBRec/. Miaomiao Cai 0001, Junfeng Fang, Zhiyong Cheng 0001, Xiang Wang 0010, Meng Wang 0001 |
WWW | 3 |
| 2026 | LPEdit: Locality-Preserving Knowledge Editing for MultiModal Large Language Models
Junfeng Fang, Houcheng Jiang, Xiang Wang 0010, Xiangnan He 0001 |
WWW | 2 |
| 2026 | Out-of-Distribution-Resistant Evaluations for Explanations of Graph Neural NetworksabstractExplainability in Graph Neural Networks (GNNs) has shown considerable promise in bolstering their trustworthiness, credibility and transparency. Our research delves into the assessment of explainability within GNNs, a pivotal factor for ensuring the reliability of explainability techniques in real-world applications. Existing evaluation metrics, typically involving taking explanatory subgraphs as inputs and measuring output differences, often face out-of-distribution (OOD) challenges. This issue occurs when explanatory subgraphs do not align with real-world data distributions, affecting the reliability of model explanations. With this in mind, in this work, we endeavor to confront this issue by introducing a novel evaluation metric, termed OOD-resistant Adversarial Robustness (OAR). Specifically, our approach is inspired by adversarial robustness, assessing the resilience of explanation subgraphs to attacks. Additionally, we incorporate a sophisticated OOD reweighting mechanism within the evaluation framework to ensure that assessments remain aligned with the original data distribution. Going beyond this, to accommodate a wider range of evaluation tasks, we further devise a counterfactual attack module and complement the perturbed subgraph using the conditional graph diffusion model. The refined paradigm, termed OAR+, ensures that our metric is versatile and applicable across various contexts. Furthermore, we establish a standardized framework, which serves as a benchmark for evaluating the fairness and accuracy of different metrics. We conduct extensive experiments to validate the effectiveness of the OAR and OAR+. Junfeng Fang, Hao Wu 0083, An Zhang 0003, Tianlong Chen 0001, Kun Wang 0056, Yuxuan Liang 0002, Roger Zimmermann, Xiang Wang 0010 |
IEEE Trans. Pattern Anal. Mach. Intell. | 1 |
| 2025 | Decoding by Contrasting Knowledge: Enhancing Large Language Model Confidence on Edited FactsabstractBaolong Bi, Shenghua Liu, Lingrui Mei, Yiwei Wang, Junfeng Fang, Pengliang Ji, Xueqi Cheng. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025. Baolong Bi, Shenghua Liu, Lingrui Mei, Yiwei Wang 0001, Junfeng Fang, Pengliang Ji, Xueqi Cheng 0001 |
ACL (1) | 5 |
| 2025 | Cracking the Code of Hallucination in LVLMs with Vision-aware Head DivergenceabstractJinghan He, Kuan Zhu, Haiyun Guo, Junfeng Fang, Zhenglin Hua, Yuheng Jia, Ming Tang, Tat-Seng Chua, Jinqiao Wang. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025. Jinghan He, Kuan Zhu, Haiyun Guo, Junfeng Fang, Zhenglin Hua, Yuheng Jia, Ming Tang 0001, Tat-Seng Chua, Jinqiao Wang |
ACL (1) | 4 |
| 2025 | Neuron-Level Sequential Editing for Large Language ModelsabstractThis work explores sequential model editing in large language models (LLMs), a critical task that involves modifying internal knowledge within LLMs continuously through multi-round editing, each incorporating updates or corrections to adjust the model’s outputs without the need for costly retraining. Existing model editing methods, especially those that alter model parameters, typically focus on single-round editing and often face significant challenges in sequential model editing-most notably issues of model forgetting and failure. To address these challenges, we introduce a new model editing method, namely Neuron-level Sequential Editing (NSE), tailored for supporting sequential model editing. Specifically, we optimize the target layer’s hidden states using the model’s original weights to prevent model failure. Furthermore, we iteratively select neurons in multiple layers for editing based on their activation values to mitigate model forgetting. Our empirical experiments demonstrate that NSE significantly outperforms current modifying parameters model editing methods, marking a substantial advancement in the field of sequential model editing. Our code is released on https://anonymous.4open.science/r/NSE-0A8D/. Houcheng Jiang, Junfeng Fang, Baolong Bi, An Zhang 0003, Xiang Wang 0010 |
ACL (1) | 2 |
| 2025 | Dynamic Multimodal Prototype Learning in Vision-Language ModelsabstractWith the increasing attention to pre-trained vision-language models (VLMs), \eg, CLIP, substantial efforts have been devoted to many downstream tasks, especially in test-time adaptation (TTA). However, previous works focus on learning prototypes only in the textual modality while overlooking the ambiguous semantics in class names. These ambiguities lead to textual prototypes that are insufficient to capture visual concepts, resulting in limited performance. To address this issue, we introduce \textbf{ProtoMM}, a training-free framework that constructs multimodal prototypes to adapt VLMs during the test time. By viewing the prototype as a discrete distribution over the textual descriptions and visual particles, ProtoMM has the ability to combine the multimodal features for comprehensive prototype learning. More importantly, the visual particles are dynamically updated as the testing stream flows. This allows our multimodal prototypes to continually learn from the data, enhancing their generalizability in unseen scenarios. In addition, we quantify the importance of the prototypes and test images by formulating their semantic distance as an optimal transport problem. Extensive experiments on 15 zero-shot benchmarks demonstrate the effectiveness of our method, achieving a 1.03\% average accuracy improvement over state-of-the-art methods on ImageNet and its variant datasets. Shuo Wang 0008, Beier Zhu, Miaoge Li, Junfeng Fang, Zhicai Wang, Dongsheng Wang 0003, Hanwang Zhang |
ICCV | 6 |
| 2025 | Is Factuality Enhancement a Free Lunch For LLMs? Better Factuality Can Lead to Worse Context-FaithfulnessabstractAs the modern tools of choice for text understanding and generation, large language models (LLMs) are expected to accurately output answers by leveraging the input context.
This requires LLMs to possess both context-faithfulness and factual accuracy.
While extensive efforts aim to reduce hallucinations through factuality enhancement methods, they also pose risks of hindering context-faithfulness, as factuality enhancement can lead LLMs to become overly confident in their parametric knowledge, causing them to overlook the relevant input context.
In this work, we argue that current factuality enhancement methods can significantly undermine the context-faithfulness of LLMs.
We first revisit the current factuality enhancement methods and evaluate their effectiveness in enhancing factual accuracy.
Next, we evaluate their performance on knowledge editing tasks to assess the potential impact on context-faithfulness.
The experimental results reveal that while these methods may yield inconsistent improvements in factual accuracy, they also cause a more severe decline in context-faithfulness, with the largest decrease reaching a striking 69.7\%.
To explain these declines, we analyze the hidden states and logit distributions for the tokens representing new knowledge and parametric knowledge respectively, highlighting the limitations of current approaches.
Our finding highlights the complex trade-offs inherent in enhancing LLMs.
Therefore, we recommend that more research on LLMs' factuality enhancement make efforts to reduce the sacrifice of context-faithfulness. Baolong Bi, Shenghua Liu, Yiwei Wang 0001, Lingrui Mei, Junfeng Fang, Hongcheng Gao, Shiyu Ni, Xueqi Cheng 0001 |
ICLR | 5 |
| 2025 | AlphaEdit: Null-Space Constrained Knowledge Editing for Language ModelsabstractLarge language models (LLMs) often exhibit hallucinations, producing incorrect or outdated knowledge. Hence, model editing methods have emerged to enable targeted knowledge updates. To achieve this, a prevailing paradigm is the locating-then-editing approach, which first locates influential parameters and then edits them by introducing a perturbation. While effective, current studies have demonstrated that this perturbation inevitably disrupt the originally preserved knowledge within LLMs, especially in sequential editing scenarios.
To address this, we introduce AlphaEdit, a novel solution that projects perturbation onto the null space of the preserved knowledge before applying it to the parameters. We theoretically prove that this projection ensures the output of post-edited LLMs remains unchanged when queried about the preserved knowledge, thereby mitigating the issue of disruption.
Extensive experiments on various LLMs, including LLaMA3, GPT2-XL, and GPT-J, show that AlphaEdit boosts the performance of most locating-then-editing methods by an average of 36.7% with a single line of additional code for projection solely. Junfeng Fang, Houcheng Jiang, Kun Wang 0056, Yunshan Ma 0002, Jie Shi 0005, Xiang Wang 0010, Xiangnan He 0001, Tat-Seng Chua |
ICLR | 1 |
| 2025 | DiffGAD: A Diffusion-based Unsupervised Graph Anomaly DetectorabstractGraph Anomaly Detection (GAD) is crucial for identifying abnormal entities within networks, garnering significant attention across various fields. Traditional unsupervised methods, which decode encoded latent representations of unlabeled data with a reconstruction focus, often fail to capture critical discriminative content, leading to suboptimal anomaly detection.
To address these challenges, we present a Diffusion-based Graph Anomaly Detector (DiffGAD). At the heart of DiffGAD is a novel latent space learning paradigm, meticulously designed to enhance the model's proficiency by guiding it with discriminative content. This innovative approach leverages diffusion sampling to infuse the latent space with discriminative content and introduces a content-preservation mechanism that retains valuable information across different scales, significantly improving the model’s adeptness at identifying anomalies with limited time and space complexity.
Our comprehensive evaluation of DiffGAD, conducted on six real-world and large-scale datasets with various metrics, demonstrated its exceptional performance. Our code is available at https://github.com/fortunato-all/DiffGAD Jinghan Li, Jinda Lu, Junfeng Fang, Congcong Wen, Xiang Wang 0010 |
ICLR | 4 |
| 2025 | NExT-Mol: 3D Diffusion Meets 1D Language Modeling for 3D Molecule Generationabstract3D molecule generation is crucial for drug discovery and material design. While prior efforts focus on 3D diffusion models for their benefits in modeling continuous 3D conformers, they overlook the advantages of 1D SELFIES-based Language Models (LMs), which can generate 100\% valid molecules and leverage the billion-scale 1D molecule datasets. To combine these advantages for 3D molecule generation, we propose a foundation model -- NExT-Mol: 3D Diffusion Meets 1D Language Modeling for 3D Molecule Generation. NExT-Mol uses an extensively pretrained molecule LM for 1D molecule generation, and subsequently predicts the generated molecule's 3D conformers with a 3D diffusion model. We enhance NExT-Mol's performance by scaling up the LM's model size, refining the diffusion neural architecture, and applying 1D to 3D transfer learning. Notably, our 1D molecule LM significantly outperforms baselines in distributional similarity while ensuring validity, and our 3D diffusion model achieves leading performances in conformer prediction. Given these improvements in 1D and 3D modeling, NExT-Mol achieves a 26\% relative improvement in 3D FCD for de novo 3D generation on GEOM-DRUGS, and a 13\% average relative gain for conditional 3D generation on QM9-2014. Our codes and pretrained checkpoints are available at https://github.com/acharkq/NExT-Mol. Zhiyuan Liu 0001, Yanchen Luo, Enzhi Zhang, Sihang Li 0002, Junfeng Fang, Yaorui Shi, Xiang Wang 0010, Kenji Kawaguchi, Tat-Seng Chua |
ICLR | 6 |
| 2025 | Iterative Substructure Extraction for Molecular Relational Learning with Interactive Graph Information BottleneckabstractMolecular relational learning (MRL) seeks to understand the interaction behaviors between molecules, a pivotal task in domains such as drug discovery and materials science. Recently, extracting core substructures and modeling their interactions have emerged as mainstream approaches within machine learning-assisted methods. However, these methods still exhibit some limitations, such as insufficient consideration of molecular interactions or capturing substructures that include excessive noise, which hampers precise core substructure extraction.
To address these challenges, we present an integrated dynamic framework called Iterative Substructure Extraction (ISE). ISE employs the Expectation-Maximization (EM) algorithm for MRL tasks, where the core substructures of interacting molecules are treated as latent variables and model parameters, respectively. Through iterative refinement, ISE gradually narrows the interactions from the entire molecular structures to just the core substructures.
Moreover, to ensure the extracted substructures are concise and compact, we propose the Interactive Graph Information Bottleneck (IGIB) theory, which focuses on capturing the most influential yet minimal interactive substructures. In summary, our approach, guided by the IGIB theory, achieves precise substructure extraction within the ISE framework and is encapsulated in the IGIB-ISE}
Extensive experiments validate the superiority of our model over state-of-the-art baselines across various tasks in terms of accuracy, generalizability, and interpretability. Junfeng Fang, Xuqiang Li, Hongxin Xiang, Alan Xia, Wenjie Du 0003, Yang Wang 0015 |
ICLR | 2 |
| 2025 | On the Role of Attention Heads in Large Language Model SafetyabstractLarge language models (LLMs) achieve state-of-the-art performance on multiple language tasks, yet their safety guardrails can be circumvented, leading to harmful generations. In light of this, recent research on safety mechanisms has emerged, revealing that when safety representations or component are suppressed, the safety capability of LLMs are compromised. However, existing research tends to overlook the safety impact of multi-head attention mechanisms, despite their crucial role in various model functionalities. Hence, in this paper, we aim to explore the connection between standard attention mechanisms and safety capability to fill this gap in the safety-related mechanistic interpretability. We propose an novel metric which tailored for multi-head attention, the Safety Head ImPortant Score (Ships), to assess the individual heads' contributions to model safety. Base on this, we generalize Ships to the dataset level and further introduce the Safety Attention Head AttRibution Algorithm (Sahara) to attribute the critical safety attention heads inside the model. Our findings show that special attention head has a significant impact on safety. Ablating a single safety head allows aligned model (e.g., Llama-2-7b-chat) to respond to **16$\times\uparrow$** more harmful queries, while only modifying **0.006\%** $\downarrow$ of the parameters, in contrast to the $\sim$ **5\%** modification required in previous studies. More importantly, we demonstrate that attention heads primarily function as feature extractors for safety and models fine-tuned from the same base model exhibit overlapping safety heads through comprehensive experiments. Together, our attribution approach and findings provide a novel perspective for unpacking the black box of safety mechanisms in large models. Zhenhong Zhou, Haiyang Yu 0003, Xinghua Zhang 0001, Rongwu Xu, Fei Huang 0002, Kun Wang 0056, Yang Liu 0005, Junfeng Fang, Yongbin Li 0001 |
ICLR | 8 |
| 2025 | Reinforced Lifelong Editing for Language ModelsabstractLarge language models (LLMs) acquire information from pre-training corpora, but their stored knowledge can become inaccurate or outdated over time. Model editing addresses this challenge by modifying model parameters without retraining, and prevalent approaches leverage hypernetworks to generate these parameter updates. However, they face significant challenges in lifelong editing due to their incompatibility with LLM parameters that dynamically change during the editing process. To address this, we observed that hypernetwork-based lifelong editing aligns with reinforcement learning modeling and proposed **RLEdit**, an RL-based editing method. By treating editing losses as rewards and optimizing hypernetwork parameters at the full knowledge sequence level, we enable it to precisely capture LLM changes and generate appropriate parameter updates. Our extensive empirical evaluation across several LLMs demonstrates that RLEdit outperforms existing methods in lifelong editing with superior effectiveness and efficiency, achieving a **59.24%** improvement while requiring only **2.11%** of the time compared to most approaches. Zherui Li 0001, Houcheng Jiang, Baolong Bi, Zhenhong Zhou, Fei Sun 0001, Junfeng Fang, Xiang Wang 0010 |
ICML | 7 |
| 2025 | AnyEdit: Edit Any Knowledge Encoded in Language ModelsabstractLarge language models (LLMs) often produce incorrect or outdated information, necessitating efficient and precise knowledge updates. Current model editing methods, however, struggle with long-form knowledge in diverse formats, such as poetry, code snippets, and mathematical derivations. These limitations arise from their reliance on editing a single token’s hidden state, a limitation we term as ``efficacy barrier''. To solve this, we propose \textbf{AnyEdit}, a new autoregressive editing paradigm. It decomposes long-form knowledge into sequential chunks and iteratively edits the key token in each chunk, ensuring consistent and accurate outputs. Theoretically, we ground AnyEdit in the Chain Rule of Mutual Information, showing its ability to update any knowledge within LLMs. Empirically, it outperforms strong baselines by 21.5\% on benchmarks including UnKEBench, AKEW, and our new \textbf{EditEverything} dataset for long-form diverse-formatted knowledge. Additionally, AnyEdit serves as a plug-and-play framework, enabling current editing methods to update knowledge with arbitrary length and format, significantly advancing the scope and practicality of LLM knowledge editing. Our code is available at: \url{https://github.com/jianghoucheng/AnyEdit}. Houcheng Jiang, Junfeng Fang, Ningyu Zhang 0001, Mingyang Wan, Guojun Ma, Xiang Wang 0010, Xiangnan He 0001, Tat-Seng Chua |
ICML | 2 |
| 2025 | DAMA: Data- and Model-aware Alignment of Multi-modal LLMsabstractDirect Preference Optimization (DPO) has shown effectiveness in aligning multi-modal large language models (MLLM) with human preferences. However, existing methods exhibit an imbalanced responsiveness to the data of varying hardness, tending to overfit on the easy-to-distinguish data while underfitting on the hard-to-distinguish data. In this paper, we propose Data- and Model-aware DPO (DAMA) to dynamically adjust the optimization process from two key aspects: (1) a data-aware strategy that incorporates data hardness, and (2) a model-aware strategy that integrates real-time model responses. By combining the two strategies, DAMA enables the model to effectively adapt to data with varying levels of hardness. Extensive experiments on five benchmarks demonstrate that DAMA not only significantly enhances the trustworthiness, but also improves the effectiveness over general tasks. For instance, on the Object HalBench, our DAMA-7B reduces response-level and mentioned-level hallucination by 90.0% and 95.3%, respectively, surpassing the performance of GPT-4V. Jinda Lu, Junkang Wu, Jinghan Li, Xiaojun Jia, Shuo Wang 0008, Yifan Zhang 0004, Junfeng Fang, Xiang Wang 0010, Xiangnan He 0001 |
ICML | 7 |
| 2025 | Multi-agent Architecture Search via Agentic SupernetabstractLarge Language Model (LLM)-empowered multi-agent systems extend the cognitive boundaries of individual agents through disciplined collaboration and interaction, while constructing these systems often requires labor-intensive manual designs. Despite the availability of methods to automate the design of agentic workflows, they typically seek to identify a static, complex, one-size-fits-all system, which, however, fails to dynamically allocate inference resources based on the difficulty and domain of each query. To address this challenge, we shift away from the pursuit of a monolithic agentic system, instead optimizing the \textbf{agentic supernet}, a probabilistic and continuous distribution of agentic architectures. We introduce \textbf{MaAS}, an automated framework that samples query-dependent agentic systems from the supernet, delivering high-quality solutions and tailored resource allocation (\textit{e.g.}, LLM calls, tool calls, token cost). Comprehensive evaluation across six benchmarks demonstrates that MaAS \textbf{(I)} requires only $6\\sim45\\%$ of the inference costs of existing handcrafted or automated multi-agent systems, \textbf{(II)} surpasses them by $0.54\\%\sim11.82\\%$, and \textbf{(III)} enjoys superior cross-dataset and cross-LLM-backbone transferability. Guibin Zhang, Luyang Niu, Junfeng Fang, Kun Wang 0056, Lei Bai 0001, Xiang Wang 0010 |
ICML | 3 |
| 2025 | G-Designer: Architecting Multi-agent Communication Topologies via Graph Neural NetworksabstractRecent advancements in large language model (LLM)-based agents have demonstrated that collective intelligence can significantly surpass the capabilities of individual agents, primarily due to well-crafted inter-agent communication topologies. Despite the diverse and high-performing designs available, practitioners often face confusion when selecting the most effective pipeline for their specific task: Which topology is the best choice for my task, avoiding unnecessary communication token overhead while ensuring high-quality solution? In response to this dilemma, we introduce G-Designer, an adaptive, efficient, and robust solution for multi-agent deployment, which dynamically designs task-aware, customized communication topologies. Specifically, G-Designer models the multi-agent system as a multi-agent network, leveraging a variational graph auto-encoder to encode both the nodes (agents) and a task-specific virtual node, and decodes a task-adaptive and high-performing communication topology. Extensive experiments on six benchmarks showcase that G-Designer is: (1) high-performing, achieving superior results on MMLU with accuracy at $84.50\%$ and on HumanEval with pass@1 at $89.90\%$; \textbf{(2) task-adaptive}, architecting communication protocols tailored to task difficulty, reducing token consumption by up to $95.33\%$ on HumanEval; and \textbf{(3) adversarially robust}, defending against agent adversarial attacks with merely $0.3\%$ accuracy drop. Guibin Zhang, Yanwei Yue, Xiangguo Sun, Guancheng Wan, Junfeng Fang, Kun Wang 0056, Tianlong Chen 0001, Dawei Cheng |
ICML | 6 |
| 2025 | ACE: Concept Editing in Diffusion Models without Performance DegradationabstractDiffusion-based text-to-image models have demonstrated remarkable capabilities in generating realistic images, but they raise societal and ethical concerns, such as the creation of unsafe content. While concept editing is proposed to address these issues, they often struggle to balance the removal of unsafe concept with maintaining the model's general generative capabilities. In this work, we propose ACE, a new editing method that enhances concept editing in diffusion models. ACE introduces a novel cross null-space projection approach to precisely erase unsafe concept while maintaining the model's ability to generate high-quality, semantically consistent images. Extensive experiments demonstrate that ACE significantly outperforms the advancing baselines, improving semantic consistency by 24.56% and image generation quality by 34.82% on average with only 1% of the time cost. These results highlight the practical utility of concept editing by mitigating its potential risks, paving the way for broader applications in the field. Junfeng Fang, Jiaqi Li 0031, Jie Shi 0005, Kun Wang 0056, Xiang Wang 0010 |
ACM Multimedia | 2 |
| 2025 | GuardReasoner-VL: Safeguarding VLMs via Reinforced ReasoningabstractTo enhance the safety of VLMs, this paper introduces a novel reasoning-based VLM guard model dubbed GuardReasoner-VL. The core idea is to incentivize the guard model to deliberatively reason before making moderation decisions via online RL.
First, we construct GuardReasoner-VLTrain, a reasoning corpus with 123K samples and 631K reasoning steps, spanning text, image, and text-image inputs.
Then, based on it, we cold-start our model's reasoning ability via SFT.
In addition, we further enhance reasoning regarding moderation through online RL.
Concretely, to enhance diversity and difficulty of samples, we conduct rejection sampling followed by data augmentation via the proposed safety-aware data concatenation.
Besides, we use a dynamic clipping parameter to encourage exploration in early stages and exploitation in later stages.
To balance performance and token efficiency, we design a length-aware safety reward that integrates accuracy, format, and token cost.
Extensive experiments demonstrate the superiority of our model.
Remarkably, it surpasses the runner-up by 19.27% F1 score on average, as shown in Figure 1.
We release data, code, and models (3B/7B) of GuardReasoner-VL: https://github.com/yueliu1999/GuardReasoner-VL. Yue Liu 0008, Shengfang Zhai, Mingzhe Du, Tri Cao, Hongcheng Gao, Xinfeng Li, Kun Wang 0056, Junfeng Fang, Jiaheng Zhang, Bryan Hooi |
NeurIPS | 10 |
| 2025 | Search and Refine During Think: Facilitating Knowledge Refinement for Improved Retrieval-Augmented ReasoningabstractLarge language models have demonstrated impressive reasoning capabilities but are inherently limited by their knowledge reservoir.
Retrieval-augmented reasoning mitigates this limitation by allowing LLMs to query external resources, but existing methods often retrieve irrelevant or noisy information, hindering accurate reasoning.
In this paper, we propose **AutoRefine**, a reinforcement learning post-training framework that adopts a new "search-and-refine-during-think" paradigm.
AutoRefine introduces explicit knowledge refinement steps between successive search calls, enabling the model to iteratively filter, distill, and organize evidence before generating an answer.
Furthermore, we incorporate tailored retrieval-specific rewards alongside answer correctness rewards using group relative policy optimization.
Experiments on single-hop and multi-hop QA benchmarks demonstrate that AutoRefine significantly outperforms existing approaches, particularly in complex, multi-hop reasoning scenarios.
Detailed analysis shows that AutoRefine issues frequent, higher-quality searches and synthesizes evidence effectively. Yaorui Shi, Sihang Li 0002, Chang Wu 0003, Zhiyuan Liu 0001, Junfeng Fang, Hengxing Cai, An Zhang 0003, Xiang Wang 0010 |
NeurIPS | 5 |
| 2025 | 3D-GSRD: 3D Molecular Graph Auto-Encoder with Selective Re-mask DecodingabstractMasked graph modeling (MGM) is a promising approach for molecular representation learning (MRL). However, extending the success of re-mask decoding from 2D to 3D MGM is non-trivial, primarily due to two conflicting challenges: avoiding 2D structure leakage to the decoder, while still providing sufficient 2D context for reconstructing re-masked atoms. To address these challenges, we propose 3D-GSRD: a 3D Molecular Graph Auto-Encoder with Selective Re-mask Decoding. The core innovation of 3D-GSRD lies in its Selective Re-mask Decoding (SRD), which re-masks only 3D-relevant information from encoder representations while preserving the 2D graph structures.
This SRD is synergistically integrated with a 3D Relational-Transformer (3D-ReTrans) encoder alongside a structure-independent decoder. We analyze that SRD, combined with the structure-independent decoder, enhances the encoder's role in MRL. Extensive experiments show that 3D-GSRD achieves strong downstream performance, setting a new state-of-the-art on 7 out of 8 targets in the widely used MD17 molecular property prediction benchmark. The code is released at https://github.com/WuChang0124/3D-GSRD. Chang Wu 0003, Zhiyuan Liu 0001, Wen Shu, Yanchen Luo, Wenqiang Lei, Yatao Bian, Junfeng Fang, Xiang Wang 0010 |
NeurIPS | 8 |
| 2025 | LIFEBENCH: Evaluating Length Instruction Following in Large Language ModelsabstractWhile large language models (LLMs) can solve PhD-level reasoning problems over long context inputs, they still struggle with a seemingly simpler task: following explicit length instructions—e.g., write a 10,000-word novel. Additionally, models often generate far too short outputs, terminate prematurely, or even refuse the request. Existing benchmarks focus primarily on evaluating generations quality, but often overlook whether the generations meet length constraints. To this end, we introduce Length Instruction Following Evaluation Benchmark (LIFEBench) to comprehensively evaluate LLMs' ability to follow length instructions across diverse tasks and a wide range of specified lengths. LIFEBench consists of 10,800 instances across 4 task categories in both English and Chinese, covering length constraints ranging from 16 to 8192 words. We evaluate 26 widely-used LLMs and find that most models reasonably follow short-length instructions but deteriorate sharply beyond a certain threshold. Surprisingly, almost all models fail to reach the vendor-claimed maximum output lengths in practice, as further confirmed by our evaluations extending up to 32K words. Even long-context LLMs, despite their extended input-output windows, counterintuitively fail to improve length-instructions following. Notably, Reasoning LLMs outperform even specialized long-text generation models, achieving state-of-the-art length following. Overall, LIFEBench uncovers fundamental limitations in current LLMs' length instructions following ability, offering critical insights for future progress. Zhenhong Zhou, Kun Wang 0056, Junfeng Fang, Rongwu Xu, Yuanhe Zhang, Xinfeng Li, Li Sun 0008, Lingjuan Lyu, Sen Su |
NeurIPS | 4 |
| 2025 | SEHG: Bridging Interpretability and Prediction in Self-Explainable Heterogeneous Graph Neural NetworksabstractHeterogeneous Graph Neural Networks (HGNNs) are extensively applied in modeling web-based applications that involve heterogeneous graph structures. Explanation models for HGNNs aim to address their ''black box'' nature. Enhancing the interpretability of HGNNs leads to a better understanding and can potentially improve predictive performance. However, existing post-hoc HGNN explanation methods cannot impact the HGNN's predictions. Self-explainable homogeneous models also perform poorly on heterogeneous graphs. To address these challenges, we present a Self-Explainable Heterogeneous Graph Neural Network (SEHG), a novel architecture that integrates explanation generation into the learning process of HGNN through two alternative stages. The first stage focuses on producing high-quality explanations while providing predictions alongside. The second stage enhances prediction accuracy by a contrastive learning strategy. Unlike the current methods that rely on manually defined metapaths for structural explanations, SEHG generates important structure and feature explanations by learnable heterogeneous masks. To ensure high-quality and sparsity explanation, these masks are regulated by a uniquely designed range-based penalty during training. Moreover, we introduce HetBA, a collection of synthetic heterogeneous datasets designed to quantify and visualize explanations or heterogeneous graphs. Extensive experiments demonstrate the effectiveness of SEHG, which surpasses strong baselines in real-world node classification tasks by notable margins of up to 3.91%. SEHG also achieves state-of-the-art performance on synthetic datasets with improvement of up to 9.44%, and records the highest fidelity scores in explanation tasks, improving by up to 46.57%. To our knowledge, SEHG is a pioneering self-explainable HGNN framework that achieves state-of-the-art performance on both heterogeneous graph explanation and prediction tasks. Zhenhua Huang 0002, Xiuyang Wu, Chengpei Xu, Junfeng Fang, Linyuan Lu, Feng Xia 0001 |
WWW | 6 |
| 2025 | Damage Analysis via Bidirectional Multi-Task Cascaded Multimodal FusionabstractDamage analysis in social media platforms such as Twitter is a comprehensive problem which involves different subtasks for mining damage-related information from tweets ( e.g., informativeness, humanitarian categories and severity assessment). The comprehensive information obtained by damage analysis enables to identify breaking events around the world in real-time and hence provides aids in emergency responses. Recently, with the rapid development of web technologies, multimodal damage analysis has received increasing attentions due to users' preference of posting multimodal information in social media. Multimodal damage analysis leverages the associated image modality to improve the identification of damage-related information in social media. However, existing works on multimodal damage analysis address each damage-related subtask individually and do not consider their joint training mechanism. In this work, we propose the Bidirectional Multi-task Cascaded multimodal Fusion (BiMCF) approach towards joint multimodal damage analysis. To this end, we introduce the cascaded multimodal fusion framework to separately integrate effective visual and text information for each task, considering that different tasks attend to different information. To exploit the interactions across tasks, bidirectional propagation of the attended image-text interactive information is implemented between tasks, which can lead to enhanced multimodal fusion. Comprehensive experiments are conducted to validate the effectiveness of the proposed approach. Code is available at https://github.com/tiggers23/BiMCF. Siying Wu, Junfeng Fang, Guowu Yang, Wenya Wang 0001, Fengmao Lv |
WWW | 3 |
| 2025 | Explainable and Efficient Editing for Large Language ModelsabstractLarge Language Models (LLMs) exhibit remarkable capabilities in storing and retrieving vast amounts of factual knowledge. However, they retain outdated or incorrect information from Web corpora. Since full retraining is costly, locate-and-edit model editing methods offer a feasible alternative. Current methods typically follow a two-stage paradigm: (1) identifying critical layers that store knowledge and (2) updating their parameters to store new knowledge. However, both phases have their inherent limitations. Firstly, layer identification is independent of the knowledge being updated, ignoring the differences in knowledge storage patterns. Secondly, parameter updating suffers from high computational overhead due to gradient descent. To solve these, we propose an Explainable and effiCient model Editing method, termed ECE. Specifically, we integrate LLM explainability into the editing process, enabling the adaptive identification of the crucial neurons. Through clustering similar knowledge, we enable batch optimization in a single gradient step, significantly reducing computational time without compromising effectiveness. Extensive experiments demonstrate that ECE can achieve superior performance, showcasing the potential of explainability-driven editing methods for LLMs. Code is available at https://github.com/tianyuzhangterry/ECE. Junfeng Fang, Houcheng Jiang, Baolong Bi, Xiang Wang 0010, Xiangnan He 0001 |
WWW | 2 |
| 2024 | Two Heads Are Better Than One: Boosting Graph Sparse Training via Semantic and Topological AwarenessabstractGraph Neural Networks (GNNs) excel in various graph learning tasks but face computational challenges when applied to large-scale graphs. A promising solution is to remove non-essential edges to reduce the computational overheads in GNN. Previous literature generally falls into two categories: topology-guided and semantic-guided. The former maintains certain graph topological properties yet often underperforms on GNNs. % due to low integration with neural network training. The latter performs well at lower sparsity on GNNs but faces performance collapse at higher sparsity levels. With this in mind, we propose a new research line and concept termed **Graph Sparse Training** **(GST)**, which dynamically manipulates sparsity at the data level. Specifically, GST initially constructs a topology & semantic anchor at a low training cost, followed by performing dynamic sparse training to align the sparse graph with the anchor. We introduce the **Equilibria Sparsification Principle** to guide this process, balancing the preservation of both topological and semantic information. Ultimately, GST produces a sparse graph with maximum topological integrity and no performance degradation. Extensive experiments on 6 datasets and 5 backbones showcase that GST **(I)** identifies subgraphs at higher graph sparsity levels ($1.67\%\sim15.85\%$$\uparrow$) than state-of-the-art sparsification methods, **(II)** preserves more key spectral properties, **(III)** achieves $1.27-3.42\times$ speedup in GNN inference and **(IV)** successfully helps graph adversarial defense and graph lottery tickets. Guibin Zhang, Yanwei Yue, Kun Wang 0056, Junfeng Fang, Yongduo Sui, Kai Wang 0036, Yuxuan Liang 0002, Dawei Cheng, Shirui Pan, Tianlong Chen 0001 |
ICML | 4 |
| 2024 | MMGNN: A Molecular Merged Graph Neural Network for Explainable Solvation Free Energy Prediction
Wenjie Du 0003, Di Wu 0057, Jun Xia 0001, Ziyuan Zhao, Junfeng Fang, Yang Wang 0015 |
IJCAI | 6 |
| 2024 | The Snowflake Hypothesis: Training and Powering GNN with One Node One Receptive FieldabstractDespite Graph Neural Networks (GNNs) demonstrating considerable promise in graph representation learning tasks, GNNs predominantly face significant issues with overfitting and over-smoothing as they go deeper as models of computer vision (CV) realm.The success of artificial intelligence in computer vision and natural language processing largely stems from its ability to train deep models effectively.We have thus conducted a systematic study on deep GNN models.Our findings indicate that the current success of deep GNNs primarily stems from (I) the adoption of innovations from CNNs, such as residual/skip connections, or (II) the tailor-made aggregation algorithms like DropEdge.However, these algorithms often lack intrinsic interpretability and indiscriminately treat all nodes within a given layer in a similar manner, thereby failing to capture the nuanced differences among various nodes.In this paper, we introduce the Snowflake Hypothesis -a novel paradigm underpinning the concept of "one node, one receptive field".The hypothesis draws inspiration from the unique and individualistic patterns of * Contribute equally to this research. Kun Wang 0056, Guohao Li 0001, Shilong Wang 0002, Guibin Zhang, Kai Wang 0036, Yang You 0001, Junfeng Fang, Xiaojiang Peng, Yuxuan Liang 0002, Yang Wang 0015 |
KDD | 7 |
| 2024 | The Heterophilic Snowflake Hypothesis: Training and Empowering GNNs for Heterophilic GraphsabstractGraph Neural Networks (GNNs) have become pivotal tools for a range of graph-based learning tasks.Notably, most current GNN architectures operate under the assumption of homophily, whether explicitly or implicitly.While this underlying assumption is frequently adopted, it is not universally applicable, which can result in potential shortcomings in learning effectiveness.In this paper, for the first time, we transfer the prevailing concept of "one node one receptive field" to the heterophilic graph.By constructing a proxy label predictor, we enable each node to possess a latent prediction distribution, which assists connected nodes in determining whether they should aggregate their associated neighbors.Ultimately, every node can have its own unique aggregation hop and pattern, much like each snowflake is unique and possesses its own characteristics.Based on observations, we innovatively introduce the Heterophily Snowflake Hypothesis and provide an effective solution to guide and facilitate research on heterophilic graphs and beyond.We conduct comprehensive experiments including (1) main results on 10 graphs with varying heterophily ratios across 10 backbones; (2) scalability on various deep GNN backbones (SGC, JKNet, etc.) across various large number of layers (2,4,6,8,16,32 layers); (3) comparison with conventional snowflake hypothesis; (4) efficiency comparison with existing graph pruning algorithms. Kun Wang 0056, Guibin Zhang, Xinnan Zhang, Junfeng Fang, Guohao Li 0001, Shirui Pan, Wei Huang 0034, Yuxuan Liang 0002 |
KDD | 4 |
| 2024 | Towards Neuron Attributions in Multi-Modal Large Language ModelsabstractAs Large Language Models (LLMs) demonstrate impressive capabilities, demystifying their internal mechanisms becomes increasingly vital. Neuron attribution, which attributes LLM outputs to specific neurons to reveal the semantic properties they learn, has emerged as a key interpretability approach. However, while neuron attribution has made significant progress in deciphering text-only LLMs, its application to Multimodal LLMs (MLLMs) remains less explored. To address this gap, we propose a novel Neuron Attribution method tailored for MLLMs, termed NAM. Specifically, NAM not only reveals the modality-specific semantic knowledge learned by neurons within MLLMs, but also highlights several intriguing properties of neurons, such as cross-modal invariance and semantic sensitivity. These properties collectively elucidate the inner workings mechanism of MLLMs, providing a deeper understanding of how MLLMs process and generate multi-modal content. Through theoretical analysis and empirical validation, we demonstrate the efficacy of NAM and the valuable insights it offers. Furthermore, leveraging NAM, we introduce a multi-modal knowledge editing paradigm, underscoring the practical significance of our approach for downstream applications of MLLMs. Junfeng Fang, Zac Bi, Houcheng Jiang, Yuan Gao 0020, Kun Wang 0056, An Zhang 0003, Jie Shi 0005, Xiang Wang 0010, Tat-Seng Chua |
NeurIPS | 1 |
| 2024 | EXGC: Bridging Efficiency and Explainability in Graph CondensationabstractGraph representation learning on vast datasets, like web data, has made significant strides. However, the associated computational and storage overheads raise concerns. In sight of this, Graph condensation (GCond) has been introduced to distill these large real datasets into a more concise yet information-rich synthetic graph. Despite acceleration efforts, existing GCond methods mainly grapple with efficiency, especially on expansive web data graphs. Hence, in this work, we pinpoint two major inefficiencies of current paradigms: (1) the concurrent updating of a vast parameter set, and (2) pronounced parameter redundancy. To counteract these two limitations correspondingly, we first (1) employ the Mean-Field variational approximation for convergence acceleration, and then (2) propose the objective of Gradient Information Bottleneck (GDIB) to prune redundancy. By incorporating the leading explanation techniques (e.g., GNNExplainer and GSAT) to instantiate the GDIB, our EXGC, the Efficient and eXplainable Graph Condensation method is proposed, which can markedly boost efficiency and inject explainability. Our extensive evaluations across eight datasets underscore EXGC's superiority and relevance. Code is available at https://github.com/MangoKiller/EXGC. Junfeng Fang, Xinglin Li, Yongduo Sui, Yuan Gao 0020, Guibin Zhang, Kun Wang 0056, Xiang Wang 0010, Xiangnan He 0001 |
WWW | 1 |
| 2024 | Graph Anomaly Detection with Bi-level OptimizationabstractGraph anomaly detection (GAD) has various applications in finance, healthcare, and security. Graph Neural Networks (GNNs) are now the primary method for GAD, treating it as a task of semi-supervised node classification (normal vs. anomalous). However, most traditional GNNs aggregate and average embeddings from all neighbors, without considering their labels, which can hinder detecting actual anomalies. To address this issue, previous methods try to selectively aggregate neighbors. However, the same selection strategy is applied regardless of normal and anomalous classes, which does not fully solve this issue. This study discovers that nodes with different classes yet similar neighbor label distributions (NLD) tend to have opposing loss curves, which we term it as "loss rivalry". By introducing Contextual Stochastic Block Model (CSBM) and defining NLD distance, we explain this phenomenon theoretically and propose a Bi-level optimization Graph Neural Network (BioGNN), based on these observations. In a nutshell, the lower level of BioGNN segregates nodes based on their classes and NLD, while the upper level trains the anomaly detector using separation outcomes. Our experiments demonstrate that BioGNN outperforms state-of-the-art methods on four benchmarks and effectively mitigates "loss rivalry". Yuan Gao 0020, Junfeng Fang, Yongduo Sui, Xiang Wang 0010, Huamin Feng, Yongdong Zhang 0001 |
WWW | 2 |
| 2024 | Invariant Graph Learning for Causal Effect EstimationabstractCausal effect estimation from networked observational data encounters notable challenges, primarily hidden confounders arising from network structure, or spillover effects that influence unit's outcomes based on neighboring treatment assignments. Existing graph neural network (GNN)-based methods have endeavored to address these challenges, utilizing the GNN's message-passing mechanism to capture hidden confounders or model spillover effects. However, they mainly focus on transductive causal effect learning on a single networked data, limiting their efficacy in inductive settings for real-world applications where networked data often originates from multiple environments influenced by potentially varying time or geographical regions. In light of this, we introduce the principle of invariance to the task of causal effect estimation on networked data, culminating in our Invariant Graph Learning (IGL) framework. Specifically, it first generates multiple networked data to simulate diverse environments from a given observational data. Then it further encourages the model to learn environment-invariant representations for confounders and spillover effects. Such a design enables the model to extrapolate beyond a single observed environment, thereby improving the performance of causal effect estimation in potential new environments. Extensive experiments on two real-world datasets demonstrates the superiority of our approach. Yongduo Sui, Caizhi Tang, Zhixuan Chu, Junfeng Fang, Yuan Gao 0020, Qing Cui, Jun Zhou 0011, Xiang Wang 0010 |
WWW | 4 |
| 2024 | Fortune favors the invariant: Enhancing GNNs' generalizability with Invariant Graph Learning
Guibin Zhang, Yiqiao Chen, Kun Wang 0056, Junfeng Fang |
Knowl. Based Syst. | 5 |
| 2024 | Modeling Spatio-Temporal Dynamical Systems With Neural Discrete Learning and Levels-of-ExpertsabstractIn this paper, we address the issue of modeling and estimating changes in the state of the spatio-temporal dynamical systems based on a sequence of observations like video frames. Traditional numerical simulation systems depend largely on the initial settings and correctness of the constructed partial differential equations (PDEs). Despite recent efforts yielding significant success in discovering data-driven PDEs with neural networks, the limitations posed by singular scenarios and the absence of local insights prevent them from performing effectively in a broader real-world context. To this end, this paper propose the universal expert module – that is, optical flow estimation component, to capture the evolution laws of general physical processes in a data-driven fashion. To enhance local insight, we painstakingly design a finer-grained physical pipeline, since local characteristics may be influenced by various internal contextual information, which may contradict the macroscopic properties of the whole system. Further, we harness currently popular neural discrete learning to unveil the underlying important features in its latent space, this process better injects interpretability, which can help us obtain a powerful prior over these discrete random variables. We conduct extensive experiments and ablations to demonstrate that the proposed framework achieves large performance margins, compared with the existing SOTA baselines. Kun Wang 0056, Hao Wu 0083, Guibin Zhang, Junfeng Fang, Yuxuan Liang 0002, Roger Zimmermann, Yang Wang 0015 |
IEEE Trans. Knowl. Data Eng. | 4 |
| 2023 | Searching Lottery Tickets in Graph Neural Networks: A Dual Perspective
Kun Wang 0056, Yuxuan Liang 0002, Pengkun Wang 0001, Xu Wang 0029, Pengfei Gu, Junfeng Fang, Yang Wang 0015 |
ICLR | 6 |
| 2023 | Evaluating Post-hoc Explanations for Graph Neural Networks via Robustness AnalysisabstractThis work studies the evaluation of explaining graph neural networks (GNNs), which is crucial to the credibility of post-hoc explainability in practical usage. Conventional evaluation metrics, and even explanation methods -- which mainly follow the paradigm of feeding the explanatory subgraph and measuring output difference -- always suffer from the notorious out-of-distribution (OOD) issue. In this work, we endeavor to confront the issue by introducing a novel evaluation metric, termed **O**OD-resistant **A**dversarial **R**obustness (OAR). Specifically, we draw inspiration from the notion of adversarial robustness and evaluate post-hoc explanation subgraphs by calculating their robustness under attack. On top of that, an elaborate OOD reweighting block is inserted into the pipeline to confine the evaluation process to the original data distribution. For applications involving large datasets, we further devise a **Sim**plified version of **OAR** (SimOAR), which achieves a significant improvement in computational efficiency at the cost of a small amount of performance. Extensive empirical studies validate the effectiveness of our OAR and SimOAR. Junfeng Fang, Wei Liu 0005, Yuan Gao 0020, An Zhang 0003, Xiang Wang 0010, Xiangnan He 0001 |
NeurIPS | 1 |
| 2023 | Cooperative Explanations of Graph Neural NetworksabstractWith the growing success of graph neural networks (GNNs), the explainability of GNN is attracting considerable attention. Current explainers mostly leverage feature attribution and selection to explain a prediction. By tracing the importance of input features, they select the salient subgraph as the explanation. However, their explainability is at the granularity of input features only, and cannot reveal the usefulness of hidden neurons. This inherent limitation makes the explainers fail to scrutinize the model behavior thoroughly, resulting in unfaithful explanations. Junfeng Fang, Xiang Wang 0010, An Zhang 0003, Xiangnan He 0001, Tat-Seng Chua |
WSDM | 1 |