Regressor-guided Diffusion Model for De Novo Peptide Sequencing with Explicit Mass Control
Authors: Shaorong Chen, Jingbo Zhou, Jun Xia
Organizations: Zhejiang University, Hangzhou, China, 310058 · AI Lab, Westlake University, Hangzhou, China, 310030 · AIMS Lab, The Hong Kong University of Science and Technology (Guangzhou), Guangzhou, China, 511453 · The Hong Kong University of Science and Technology, Hong Kong, China, 999077
The discovery of novel proteins relies on sensitive protein identification, for which de novo peptide sequencing (DNPS) from mass spectra is a crucial approach. While deep learning has advanced DNPS, existing models inadequately enforce the fundamental mass consistency constraint, that a predicted peptide's mass must match the experimental measured precursor mass. Previous DNPS methods often treat this critical information as a simple input feature or use it in post-processing, leading to numerous implausible predictions that do not adhere to this fundamental physical property. To address this limitation, we introduce DiffuNovo, a novel regressor-guided diffusion model for de novo peptide sequencing that provides explicit peptide-level mass control. Our approach integrates the mass constraint at two critical stages: during training, a novel peptide-level mass loss guides model optimization, while at inference, regressor-based guidance from gradient-based updates in the latent space steers the generation to compel the predicted peptide adheres to the mass constraint. Comprehensive evaluations on established benchmarks demonstrate that DiffuNovo surpasses state-of-the-art methods in DNPS accuracy. Additionally, as the first DNPS model to employ a diffusion model as its core backbone, DiffuNovo leverages the powerful controllability of diffusion architecture and achieves a significant reduction in mass error, thereby producing much more physically plausible peptides. These innovations represent a substantial advancement toward robust and broadly applicable DNPS. The source code is available in the supplementary material.
Figures & tables
Figure 1: Schematic of a typical bottom-up proteomics workflow ( Zhang et al. 2013 ) . Proteins are digested into peptides and analyzed by mass spectrometer to produce mass spectrum. DNPS is the process of inferring a peptide’s sequence directly from its spectrum. The resulting peptide sequences can then be used for protein sequence assembly.
Figure 2: Performance comparison of DiffuNovo with state-of-the-art methods on mass-related metrics. Plausible prediction rate is the percentage of predicted peptides whose mass satisfies the experimental mass (higher is better). Mass error is the mean absolute error between the predicted peptide mass and the experimental mass (lower is better).
Figure 3: The architecture of DiffuNovo. In the Forward Diffusion Process, a ground-truth peptide sequence is converted into an embedding z0 and then corrupted by adding noise over T timesteps to produce a noisy latent variable zT . DiffuNovo is trained to reverse this process. The Spectrum Encoder transforms the mass spectrum into embedding x . The core of DiffuNovo is Reverse Diffusion Process where the Peptide Decoder denoise the noisy latent zt into a cleaner latent zt−1 conditioned on x . Critically, the Peptide Mass Regressor provides guidance to the Peptide Decoder at each timestep t , compelling the prediction adheres to the mass constraint. Finally, the clean latent z^0 is passed to a prediction head to output the final predicted peptide.
z0∼qϕ(z0∣y)=N(z0;Embϕ(y),(1−α0)I)
(12)
Algorithm 1 Training Stage of DiffuNovo.
μθ(zti,ti,x)←λ1zti+λ2gθ(zti,ti,x)
(15)
Algorithm 2 Inference Process of DiffuNovo.
Figure 4: Performance comparison on mass-related metrics.
Method
HelixNovo
CasaNovo
DiffuNovo
MAE( ↓ )
21.388
8.583
2.251
Analysis of DiffuNovo’s Improvement
DiffuNovo vs.
HelixNovo
CasaNovo
DiffuNovo
X% Decrease
89.5%
73.8%
0%
Fold Decrease
9.50 ×
3.81 ×
1.00 ×
Table 1: Evaluation of mass error. We compares DiffuNovo with two baselines and reports the Mean Absolute Error (MAE), where lower is better. The upper section shows the mass errors. The lower section quantifies this improvement.
Peptide-level Performance
Amino Acid-level Performance
Models
9-species
HC-PT
7-species
9-species
HC-PT
7 species
Prec.
AUC
Prec.
AUC
Prec.
AUC
Prec.
Recall
Prec.
Recall
Prec.
Recall
DeepNovo
0.428
0.376
0.313
0.255
0.204
0.136
0.696
0.638
0.531
0.534
0.492
0.433
PointNovo
0.480
0.436
0.419
0.373
0.022
0.007
0.740
0.671
0.623
0.622
0.196
0.169
CasaNovo
0.481
0.439
0.211
0.177
0.119
0.084
0.697
0.696
0.442
0.453
0.322
0.327
AdaNovo
0.505
0.469
0.212
0.178
0.174
0.135
0.698
0.709
0.442
0.451
0.379
0.385
Table 2: The comparison of de novo peptide sequencing performance between our proposed model, DiffuNovo, and other state-of-the-art methods on the three benchmark datasets. We report precision and AUC at the peptide level, and precision and recall at the amino acid level. The best and the second best are highlighted with bold and underline , respectively.
PTM Precision
Models
Nine-Species
HC-PT
Seven-Species
DeepNovo
0.576
0.626
0.391
PointNovo
0.629
0.676
0.117
AdaNovo
0.652
0.552
0.448
CasaNovo
0.706
0.501
0.360
HelixNovo
0.680
0.568
0.473
Table 3: Empirical comparison of PTM identification. We evaluate the ability of DiffuNovo and other models to identify Post-Translational Modifications (PTMs) on the benchmark datasets. The best results and the second best are highlighted with bold and underline , respectively.
Figure 5: Mass error loss curves. The figure illustrates the mass error loss (MSE) as a function of the normalized diffusion timestep. We conducted experiment on 5 random batches (2560 samples), calculating and plotting the loss for each batch (dashed line) and the average loss (solid line).
Mass spectrometry measures ion mass-to-charge patterns to identify and quantify molecules, with AI linking spectra to structures, peptides, and biological signals.
De novo peptide sequencing from tandem mass spectrometry is pivotal in proteomics, enabling identification of novel peptides without reference databases. While recent Transformer-based encoder-decoder models have achieved remarkable performance, we uncover a critical pathology in their inference dynamics. Through comprehensive feature scaling experiments, we demonstrate that existing auto-regressive peptide decoders tend to over-rely on generated-sequence priors while progressively under-utilizing fine-grained physical evidence from the input mass spectrum. This phenomenon leads to suboptimal results, where generated peptide sequences are biologically plausible yet not faithful to the input spectrum. To rectify this, we propose MemNovo, a training-free and plug-and-play mechanism that re-balances peptide and spectral contributions at inference time. MemNovo alleviates the information bottleneck by establishing a persistent spectral memory bank and injecting retrieved features directly into the final decoding stage via an ultra-conservative residual connection. Theoretical analysis confirms that this mechanism restores the mutual information between the decoder state and the raw spectrum. Extensive experiments on the Nine Species benchmark with two representative baselines, Casanovo and InstaNovo, demonstrate that MemNovo consistently improves both amino acid precision and peptide precision, achieving up to 39.1% relative improvement in peptide precision for Casanovo and up to 3.9% for InstaNovo, with negligible computational overhead.
Dongxin Lyu, Jingbo Zhou, Hongxin Xiang +2
Westlake University Hangzhou, Zhejiang, China · Hunan University Changsha, Hunan, China · Shanghai Artificial Intelligence Laboratory Shanghai, China +1
De novo peptide sequencing from tandem mass spectra is essential for identifying peptides without relying on reference databases. Despite advances in deep learning, accurate sequencing remains challenging because experimental spectra are often sparse, noisy, and incomplete, leaving informative b- and y-ion fragments unobserved. Existing methods attempt to recover this missing evidence via latent-space imputation before autoregressive decoding. However, they typically treat imputation as a fixed reconstruction task, without considering which missing fragments are most relevant to decoder errors. Moreover, existing peak representations do not explicitly model mass differences between peaks, despite their fundamental importance. We introduce GyroNovo, a framework with two main contributions. First, we use decoder errors observed during training to adapt the imputation objective, prioritizing fragments associated with frequent decoding errors. We further use the decoder error distribution to construct easy and hard augmented views of each spectrum, enabling the decoder to learn under varying degrees of spectral corruption and missing-fragment severity. Second, we introduce a mass-aware inductive bias into self-attention by using rotary embeddings to encode pairwise mass differences between spectral peaks. Together, these components align missing-fragment recovery with decoder behavior while explicitly incorporating the mass relationships that underlie peptide fragmentation. At inference time, GyroNovo retains a standard encoder-imputer-decoder architecture and requires neither additional inputs nor auxiliary search procedures. Experiments on NovoBench show gains of about 9 percentage points in peptide-level precision and 7 percentage points in amino-acid-level precision over the state-of-the-art baseline. Code: https://github.com/UBC-NLP/gyronovo.
Abdellah El Mekki, Laks V. S. Lakshmanan, Muhammad Abdul-Mageed
The University of British Columbia · Canada Research Chair in NLP and ML
We introduce a novel invariance for peptide tandem mass spectrometry data, unlocking self-supervised representation learning that improves de novo sequencing of peptides. This invariance exploits the physical relationship between precursor properties (mass and charge) and fragment-ion evidence, without requiring peptide sequence labels. We introduce dIon, which adapts the DINO framework with two latent prediction tasks, both recovering a clean teacher representation: one from a spectrum mixture, using the precursor as a selection query, and one from a partial spectrum with the precursor withheld. The first associates precursor information with fragment-ion evidence; the second prevents representational collapse onto that information alone. Mechanistic probes support both effects, and ablations show that the full objective performs best. Under identical end-to-end training, dIon initialization improves de novo peptide precision over training from scratch by 5.5 and 8.4 percentage points on the held-out MassIVE-KB and Kingdoms test sets, and by 2.3 and 4.8 percentage points with a larger supervised training corpus. The resulting models surpass fully supervised state-of-the-art de novo sequencing models on the diverse, multi-species Kingdoms corpus under the same greedy-decoding protocol. Without peptide labels, dIon learns strong native peptide-similarity geometry compared with other learned models; with limited peptide-supervised adaptation, it achieves the best retrieval and pair-discrimination performance across all representation benchmarks.
Alfred Nilsson, Joel Lapin, Samuel H. Payne +2
Science for Life Laboratory, KTH Royal Institute of Technology Stockholm, Sweden · Computational Mass Spectrometry, TUM School of Life Sciences Technical University of Munich, Freising, Germany · Biology Department, Brigham Young University Provo, Utah 84602, United States +1