← Search

Kangjie Zheng

7 accepted papers

2026

Compactness and Consistency: A Conjoint Framework for Deep Graph Clustering

ICLR 2026oral

Graph clustering is a fundamental task in data analysis, aiming at grouping nodes with similar characteristics in the graph into clusters. This problem has been widely explored using graph neural networks (GNNs) due to their ability to leverage node attributes and graph topology for effective cluste…

Cited by 0SourcecodeScholar
2025

ExLM: Rethinking the Impact of $\texttt{[MASK]}$ Tokens in Masked Language Models

ICML 2025poster

Masked Language Models (MLMs) have achieved remarkable success in many self-supervised representation learning tasks. MLMs are trained by randomly masking portions of the input sequences with $\texttt{[MASK]}$ tokens and learning to reconstruct the original content based on the remaining context. Th…

Cited by 0SourcePDFScholar
2025

SMI-Editor: Edit-based SMILES Language Model with Fragment-level Supervision

ICLR 2025poster

SMILES, a crucial textual representation of molecular structures, has garnered significant attention as a foundation for pre-trained language models (LMs). However, most existing pre-trained SMILES LMs focus solely on the single-token level supervision during pre-training, failing to fully leverage…

Cited by 1SourcePDFScholar
2024

A Decoding Algorithm for Length-Control Summarization Based on Directed Acyclic Transformers

EMNLP 2024finding

Length-control summarization aims to condense long texts into a short one within a certain length limit. Previous approaches often use autoregressive (AR) models and treat the length requirement as a soft constraint, which may not always be satisfied. In this study, we propose a novel length-control…

2024

ESM All-Atom: Multi-Scale Protein Language Model for Unified Molecular Modeling

ICML 2024poster

Protein language models have demonstrated significant potential in the field of protein engineering. However, current protein language models primarily operate at the residue scale, which limits their ability to provide information at the atom level. This limitation prevents us from fully exploiting…

2024

Mol-AE: Auto-Encoder Based Molecular Representation Learning With 3D Cloze Test Objective

ICML 2024poster

3D molecular representation learning has gained tremendous interest and achieved promising performance in various downstream tasks. A series of recent approaches follow a prevalent framework: an encoder-only model coupled with a coordinate denoising objective. However, through a series of analytical…

Cited by 7SourcePDFScholar
2023

Towards a Unified Training for Levenshtein Transformer

ICASSP 2023accepted

Levenshtein Transformer (LevT) is a widely-used text-editing model, which generates a sequence based on editing operations (deletion and insertion) in a non-autoregressive manner. However, it is challenging to train the key refinement components of LevT due to training-inference discrepancy. By care…

Cited by 0SourceScholar