AI + Vision & LanguageResearch projectBiomedical NLP

Teaching a model to match biomedical questions to answers by meaning.

Five students built a semantic similarity model for biomedical question answering. A PubMedBERT teacher scored about 6,000 BioASQ question–snippet pairs and a BioLinkBERT student learned from it. Their adjusted dual-loss training held steady through epochs 4–5, while the baseline started declining after epoch 3. They also found that snippets overlapped across their data splits, which likely inflated their scores, and said so.

ConcentrationAI + Vision & Language
FormatPBL poster
When2025 · Team of 5
Project LeadResearcher in Machine Learning and NLP, Concordia University
Pairs~6,000PubMedBERT teacher1 · Label pairsModelDual lossBioLinkBERT2 · Train studentReviewLimitsepochs, leakage3 · Check honestly
Illustration of the team's method, drawn from their poster.
~6,000Question–snippet pairs from BioASQ Task 13b
0.66%Negative pairs the teacher scored above 0.5
Epochs 4–5Where the adjusted model performed best
Abstract

Keyword search misses biomedical answers that use different words for the same idea. This team built a semantic textual similarity (STS) system for biomedical question answering using a teacher–student setup. PubMedBERT, pretrained on PubMed text, scored question–snippet pairs built from BioASQ Task 13b (about 6,000 pairs); only 0.66% of negative pairs scored above 0.5, so the teacher separated pairs well. BioLinkBERT was then trained as the student. A baseline using cosine similarity with MSE gave unstable gradients and declined after epoch 3. The team's adjusted dual loss (SmoothL1 plus an alignment loss with adaptive weighting) kept Pearson and Spearman correlations stable through epochs 4–5; extending to epoch 8 showed both declined after epoch 5. They reported that snippets overlapped across their training, validation and test sets, likely inflating the scores, and called for stricter deduplication.

Keywords · semantic textual similarity · biomedical NLP · PubMedBERT · BioLinkBERT · knowledge distillation · BioASQ

The team

Who did this work

YZSwati RajeshJiwoo HongMAShuyi Li
5 students · 5 universities · 4 countries
YZ
Yuhang Zhou
Shibaura Institute of Technology · Japan
Swati Rajesh
Swati Rajesh
National University of Singapore · Singapore
Jiwoo Hong
Jiwoo Hong
Korea University · South Korea
MA
Mariam Alamoodi
United Arab Emirates University · United Arab Emirates
Shuyi Li
Shuyi Li
The University of Tokyo · Japan

Students named here agreed to be featured.

Background

1Why keyword search falls short in biomedicine

Biomedical questions and answers often use different terms for the same concept, so keyword search misses relevant evidence.

Most sentence-similarity models are trained on general text such as Wikipedia. The team wanted a model that measures similarity the way a biomedical reader would.

Method

2A domain teacher and a student model

  1. Built the dataset. Paired BioASQ Task 13b questions with ideal answers and supporting snippets, and made negative pairs by mismatching them.
  2. Labeled with a teacher. Used PubMedBERT to score each pair; only 0.66% of negative pairs scored above 0.5.
  3. Trained a student. Fine-tuned BioLinkBERT to reproduce and refine the teacher's scores.
  4. Fixed unstable training. Replaced a cosine + MSE loss with SmoothL1 plus an alignment loss and adaptive weighting.
Results

3More stable training, and an honest caveat

Baseline dual lossAdjusted dual loss
LossCosine similarity + MSE, fixed weightsSmoothL1 + alignment loss, adaptive weights
Pearson and Spearman over epochsDeclined after epoch 3Stable through epochs 4–5, declined after 5
Training behaviorOscillating gradientsMore stable, delayed overfitting
Table 1. From the team's poster.
Reported, not hidden. The team found that snippets overlapped across training, validation and test sets, which likely inflated their correlation scores. They named stricter snippet-level deduplication as the next step.
From the team's poster

4The poster

The team's final poster.

Team poster: enhancing biomedical literature retrieval with semantic textual similarity
Figure 1. The team's poster: method, training curves and conclusions.
Limitations

5What this does not show

  • Scores are likely inflated by overlap between data splits.
  • Labels come from a teacher model, not from human annotators.
  • Results are not compared with public STS benchmarks.
Skills and fit

6What this work shows

Transformer fine-tuningKnowledge distillationLoss-function designBiomedical NLPData-leakage checksTeam research
Project Lead
PL

Researcher in Machine Learning and NLP, Concordia University

Researches implicit discourse with language models. Four peer-reviewed papers. Built NLP models for the Government of Québec.

The team built the model and ran the experiments. The Project Lead guided the work and reviewed it.

Good fit for applications in

Natural language processing · Biomedical informatics

A model with a stated data problem and a plan to fix it reads as mature research.

Project Lead review

7What the Project Lead observed

The Project Lead wrote feedback on the project at the end of the program.

Rating

Written feedback only

This cohort ran before the four-level capability rating was recorded, so this page shows the Project Lead's written feedback without a level.

Method
Strong methodological rigor: each architecture and training decision was backed by systematic experiments and clear analysis.
Understanding
A solid understanding of model behavior, with limitations addressed thoughtfully.
Next steps
Realistic next steps for improvement.
Overall
A well-executed example of applied semantic modeling in a biomedical setting.

Source: the Project Lead's and Academic Team's written feedback, 2025.

References

References

  1. Gu, Y. et al. (2021). Domain-specific language model pretraining for biomedical natural language processing. ACM Transactions on Computing for Healthcare, 3(1).
  2. Yasunaga, M. et al. (2022). LinkBERT: Pretraining language models with document links. ACL.
  3. Reimers, N. & Gurevych, I. (2019). Sentence-BERT: Sentence embeddings using Siamese BERT-networks. EMNLP.
  4. Tsatsaronis, G. et al. (2015). An overview of the BIOASQ large-scale biomedical semantic indexing and question answering competition. BMC Bioinformatics, 16.
Hong, J., Li, S., Rajesh, S., Alamoodi, M. & Zhou, Y. (2025). Enhancing biomedical literature retrieval with semantic textual similarity. AI in Natural Language Processing – Hugging Face Project, PBL poster. blendedlearn.org/outcome-biomedical-semantic-similarity
Build work like this

Join the AI in Natural Language Processing – Hugging Face Project

Work with an NLP researcher on language models and leave with research you can explain.