Comparative analysis of adaptive immune repertoires at population scale is hampered by two practical bottlenecks: the near-quadratic cost of pairwise affinity evaluations and dataset imbalances that obscure clinically important minority clonotypes. We introduce SubQuad, an end-to-end pipeline that addresses these challenges by combining antigen-aware, near-subquadratic retrieval with GPU-accelerated affinity kernels, learned multimodal fusion, and fairness-constrained clustering. The system employs compact MinHash prefiltering to sharply reduce candidate comparisons, a differentiable gating module that adaptively weights complementary alignment and embedding channels on a per-pair basis, and an automated calibration routine that enforces proportional representation of rare antigen-specific subgroups. On large viral and tumor repertoires SubQuad achieves measured gains in throughput and peak memory usage while preserving or improving recall@k, cluster purity, and subgroup equity. By co-designing indexing, similarity fusion, and equity-aware objectives, SubQuad offers a scalable, bias-aware platform for repertoire mining and downstream translational tasks such as vaccine target prioritization and biomarker discovery.
Figures & tables
Figure 1: Overview of the SubQuad framework for near-quadratic-free, equity-aware repertoire inference. Scalable Preprocessing : Raw sequences S are processed via MinHash-based Indexing to generate a sparse candidate list CAND and optimized using hardware-aware batching B . Representation Learning : A Dual-Phase Meta-Encoder utilizes ImmunoBERT -style pretraining followed by MetaNet fine-tuning. The Meta-Controller dynamically adjusts gating weights αm for multi-paradigm fusion. Graph Construction : Multi-channel affinities are integrated via Dynamic Affinity Fusion to produce aij . This similarity matrix is refined through RMT-based Thresholding (eigenvalue spectrum analysis) to produce a sparse weighted graph G=(V,E,W) . Fairness-Constrained Clustering : The graph is partitioned into clusters C by optimizing a joint objective of spatial cohesion and Jensen-Shannon Equity . An Automated Fairness Tuner dynamically calibrates the trade-off weight λ to meet target disparity δmax . Outputs : The pipeline yields antigen-aware clusters, topological maps (UMAP), and equity heatmaps for clinical interpretation.
Figure 2: Community structure in immune receptor networks. Vertices denote unique CDR3 β sequences, sized by clonal frequency and colored by primary antigen. Edges connect receptors with fused similarity above 0.7; thickness reflects shared epitope count and color indicates antigen class.
Tool (Year)
Throughput
Recall
Memory
Purity
Equity Score
(k seq/s)
(AUC)
(GB)
(%)
SubQuad (Ours)
97.2
0.985
1.4
92
0.91
BertTCR [ 30 ]
84.5
0.970
2.1
87
0.83
TCR-pMHC (PyG) [ 31 ]
60.0
0.920
3.5
82
0.78
ProtBert [ 32 ]
62.3
0.940
3.8
79
0.75
HeteroTCR [ 33 ]
75.0
0.950
1.6
85
0.79
Table 1: Comprehensive Performance Comparison of TCR Analysis Tools (10K Sequences). † All improvements ≥3% are significant at p<0.01 under paired bootstrap (10 000 resamples).
Figure 5: F1 Score Heatmap for MinHash Parameter Selection
Figure 6: Parameter optimization landscape for MinHash configurations.
Figure 7: Performance enhancement across computational domains.
Figure 8: Topological community organization in immune receptor network. Node size indicates TCR frequency, node color indicates primary antigen, edge thickness represents the number of shared antigens, and edge color denotes shared antigen type. Key pathways are highlighted for CMV-specific TCR convergence (orange), EBV-associated immunotherapy targets (green), and cross-reactive pathways (blue).
Figure 9: Feature distributions of immune receptor sequences across different antigens. Each subplot compares two features among frequency, CDR3 length, centrality, community, and eigen centrality. Colors indicate antigen specificity.
Figure 10: Scale sensitivity of fairness metrics. Normalized disparity ( Ψ=Δfair/Δbaseline ) is plotted for JS divergence, demographic parity, and equalized odds across sequence scales. The green shaded region highlights the stability of equalized odds for large-scale settings.
Figure 11: Clinical decision support dashboard with human-AI collaboration.
Model
Antigen Macro-F1
Rare Recall@100
AUPRC
ESM-2-150M [ 41 ]
0.627±0.011
0.481±0.018
0.601
ProtST-ESM-1B [ 42 ]
0.645±0.009
0.503±0.015
0.618
SubQuad (Ours)
0.712±0.006
0.594±0.010
0.681
Appendix
Table 7: Performance comparison on antigen classification and rare clonotype retrieval tasks.
Foundation models are increasingly adapted for downstream medical imaging tasks, yet the influence of the chosen adaptation strategy on subgroup fairness remains poorly understood. We investigate how three parameter-efficient adaptation techniques, including linear heads on the raw CLS token, an MLP, and an attention-pooling module over multi-layer patch features, affect both pathology classification performance and subgroup disparities when applied to the frozen Rad-DINO chest X-ray encoder. Using MIMIC-CXR, we evaluate eight pathologies across race, sex, and imaging-view subgroups on a prevalence-preserving, demographically balanced test set, and additionally probe how strongly each adapter encodes protected attributes. We find that attention pooling achieves the strongest overall discriminative performance and encodes attributes, particularly race, most strongly, but that improved overall performance does not consistently reduce subgroup disparities. Notably, stronger attribute encoding did not correspond to larger disparities: early network layers encoded race most weakly yet produced the largest subgroup performance gaps. Exploring different attention-pooling layer combinations further revealed no consistent relationship between the layers pooled, attribute encoding strength, and subgroup fairness. Our results indicate that richer, more expressive representations can improve accuracy while leaving fairness implications task-dependent and unpredictable, which must be assessed directly and per-task rather than inferred from encoding strength or overall performance alone.
Dhruv Gupta, Emma A. M. Stanley, Fabio De Sousa Ribeiro +2
Department of Computing, Imperial College London, UK · Causality in Healthcare AI Hub, UK · Department of Radiology, Royal Brompton Hospital, UK +1
Recovering the underlying k-clustering of a set U of n points by asking pair-wise same-cluster queries has garnered significant interest in the past few years. Given a query S⊂U, ∣S∣=2, the oracle returns "yes" if the points are in the same cluster and "no" otherwise. For adaptive algorithms, the query complexity is known to be Θ(nk), while non-adaptive algorithms are extremely limited: even for k=3, such algorithms require Ω(n2) queries, matching the trivial upper bound. However, non-adaptivity is highly desirable since it allows queries to be asked in parallel. To break the quadratic barrier for non-adaptive queries, we study a natural generalization of this problem to subset queries for ∣S∣>2, where the oracle returns the number of clusters intersecting S. Previous work obtained an O(n) query adaptive algorithm, but the realm of non-adaptive algorithms remained completely unknown. In this paper, we give the first non-adaptive algorithms for clustering with subset queries. Our main result is a non-adaptive algorithm making O(nlogk⋅(logk+loglogn)2) queries, improving to O(nloglogn) when k is constant. In addition to non-adaptivity, we make other practical considerations, such as enforcing a bound, s, on the query size. We show Ω(max(n2/s2,n)) queries are necessary and obtain algorithms making O(n2k/s2) queries for any s≤n and O(n2/s) queries for any s≤n. Finally, we obtain improved upper bounds when the clusters are roughly balanced, and when the algorithm is allowed two rounds of adaptivity.
Hadley Black, Euiwoong Lee, Arya Mazumdar +1
CUNY Baruch College · University of Michigan · UC San Diego
Accurate prediction of antibody-antigen binding affinity is fundamental to therapeutic design, yet remains constrained by severe label sparsity and the complexity of antigenic variations. In this paper, we propose AbLWR (Antibody-antigen binding affinity List-Wise Ranking), a novel framework that reformulates the conventional affinity regression task as a listwise ranking problem. To mitigate label sparsity, AbLWR incorporates a PU (Positive-Unlabeled) learning mechanism leveraging a dual-level contrastive objective and meta-optimized label refinement to learn robust representations. Furthermore, we address antigenic variation by employing a homologous antigen sampling strategy where Multi-Head Self-Attention (MHSA) explicitly models inter-sample relationships within training lists to capture subtle affinity nuances. Extensive experiments demonstrate that AbLWR significantly outperforms state-of-the-art baselines, improving the Precision@1 (P@1) by over 10% in randomized cross-validation experiments. Notably, case studies on Influenza and IL-33 validate its practical utility, demonstrating robust ranking consistency in distinguishing subtle viral mutations and efficiently prioritizing top-tier candidates for wet-lab screening.
Fan Xu, Zhi-an Huang, Haohuai He +5
Department of Data Science and Artificial Intelligence, The Hong Kong Polytechnic University, Hong Kong SAR, China · Department of Computer Science, City University of Hong Kong (Dongguan), Dongguan, China · School of Computer Science and Engineering, Sun Yat-sen University, Guangzhou, Guangdong, China +1