Zurück zum Blog

Embeddings für die klinische Kodierung: allgemein oder spezifisch?

Ein erfahrener Arzt

Eine Tietai-Studie zeigt, dass ein spanischer biomedizinischer Bi-Encoder mit Cross-Encoder-Reranking öffentliche Embeddings und BM25 bei der Suche nach ICD-10- / CIE-10-ES-Codes übertrifft.

Forschungshinweis. Diese Studie evaluiert Retrieval-Modelle an öffentlichen Benchmarks. Sie beschreibt kein verfügbares Produkt und nicht dessen klinische Leistung.

Klinische Erwähnung

Eine kurze Anfrage auf EN, ES, CA, IT, PT oder FR

Stufe 1 · Bi-Encoder (mehrsprachig)

Kodiert die Anfrage und jede CIE-10-Beschreibung unabhängig und rankt sie nach Kosinus-Ähnlichkeit. Trainiert mit 19.502 kuratierten synthetischen Beispielen aus 17 ICD-10-Kapiteln.

Top-k-Kandidaten

Stufe 2 · Cross-Encoder-Reranker (Spanisch)

Liest jedes Anfrage-Code-Paar gemeinsam und ordnet die Auswahlliste neu. Trainiert mit 10.628 Listwise-Gruppen mit klinisch plausiblen Hard Negatives.

Gerankte CIE-10-ES-Codes

Seite, Schweregrad, Ätiologie und Episodentyp in der finalen Reihenfolge aufgelöst

Zweistufiges ICD-Retrieval. Ein mehrsprachiger Bi-Encoder ruft CIE-10-Kandidatencodes ab, dann ordnet ein auf Spanisch abgestimmter Cross-Encoder die Auswahlliste neu, um klinisch relevante Unterscheidungen wie Seitenlokalisation, Schweregrad, Ätiologie und Episodentyp wiederherzustellen.

01 - Die Frage

Können kompakte, aufgabenspezifische Retriever breite öffentliche Embeddings übertreffen?

Satz-Embeddings funktionieren gut für breite semantische Suche, aber ICD-Kodierung ist keine breite semantische Suche. Das Modell muss kurze klinische Erwähnungen einem kontrollierten Vokabular zuordnen, in dem sich benachbarte Codes oft nur durch einen Qualifikator unterscheiden: anatomische Lokalisation, Seite, Schweregrad, Kontaktart oder Ätiologie.

Die Arbeit fragt, ob hochwertige synthetische Supervision diese Lücke für die spanische klinische Kodierung schließen kann. Wir nutzen ein Frontier-LLM, um mehrsprachige Beispiele auf Basis der ICD-10-Hierarchie zu erzeugen, stimmen einen spanischen biomedizinischen Bi-Encoder fein ab und trainieren dann einen Cross-Encoder-Reranker auf Listwise-Gruppen mit klinisch plausiblen Hard Negatives.

Das zentrale Ergebnis: Für das spanische ICD-10/CIE-10-ES-Retrieval zählen aufgabengerechte Supervision und klinisch sinnvolle Negativbeispiele mehr als Embedding-Größe oder generisches englisches biomedizinisches Pretraining.

02 - Methode

Ein Bi-Encoder für den Recall, ein Cross-Encoder für die Präzision beim exakten Code.

Der Retriever nutzt eine standardmäßige, aber sorgfältig abgestimmte zweistufige Architektur. Stufe 1 kodiert jede Anfrage und jede CIE-10-Beschreibung unabhängig und rankt nach Kosinus-Ähnlichkeit. Stufe 2 liest jedes Anfrage-Code-Paar gemeinsam und ordnet die besten Kandidaten mit einem Listwise-Ziel neu.

Die Trainingsdaten wurden nicht als rohe synthetische Ausgabe behandelt. Wir haben die erzeugten Beispiele durch regelbasierte Validierung, semantische Deduplizierung, Sprachkonsistenzprüfungen und manuelle Stichproben kuratiert, bevor wir damit den Bi-Encoder und die Reranking-Klassifikatoren gebaut haben. Diese Kuratierung ist zentral für die Arbeit: Es geht nicht einfach darum, mehr Daten zu erzeugen, sondern Beispiele und Hard Negatives zu schaffen, die die in ICD-10/CIE-10 kodierten klinischen Unterscheidungen widerspiegeln.

Synthetische Bi-Encoder-Beispiele aus 17 ICD-10-Kapiteln.
19.502
Für die synthetische Supervision verwendete Sprachen: EN, ES, CA, IT, PT und FR.
6
Spanische Listwise-Gruppen für den Cross-Encoder mit einem Positivbeispiel und Hard Negatives.
10.628
Evaluierungsebenen: vollständiger Code, dreistellige Kategorie und Kapitel.
3

Verglichene Retrieval-Strategien

StrategieRolle in der Studie
Tietai Cross-EncoderOrdnet die Auswahlliste des Bi-Encoders neu und optimiert die Reihenfolge des exakten Codes.
Tietai Bi-EncoderSpanisches biomedizinisches Dense-Retrieval-Rückgrat, feinabgestimmt mit synthetischer Supervision.
BM25 (Postgres FTS)Lexikalische Volltext-Baseline über die CIE-10-ES-Beschreibungen.
ST MiniLM-L6-v2Öffentliche Allzweck-Sentence-Transformer-Baseline.
ST BioBERTEnglische biomedizinische Sentence-Transformer-Baseline.
ST MPNet-v2Größere allgemeine Sentence-Transformer-Baseline mit 768 Dimensionen.

03 - Ergebnisse

F1 und MAP@10 rücken die Ranking-Qualität in den Fokus.

Auf CodiESP v4 erreicht der Tietai-Cross-Encoder F1 = 0,709 und MAP@10 = 0,747 auf Ebene des exakten Codes. Auf DISTEMIST erreicht er F1 = 0,776 und MAP@10 = 0,812. Die aktualisierte Fassung der Arbeit verwendet MAP@10 als Ranking-Metrik und F1 als Top-1-Entscheidungsmetrik.

F1 für den exakten Code auf CodiESP für den Tietai Cross-Encoder.
0,709
MAP@10 für den exakten Code auf CodiESP für den Tietai Cross-Encoder.
0,747
F1 für den exakten Code auf DISTEMIST für den Tietai Cross-Encoder.
0,776
MAP@10 für den exakten Code auf DISTEMIST für den Tietai Cross-Encoder.
0,812

F1 und MAP@10 auf CodiESP v4

ModellF1 exaktF1 KategorieMAP@10 exaktMAP@10 Kategorie
Tietai Cross-Encoder0,7090,8230,7470,851
Tietai Bi-Encoder0,3590,6170,4610,694
BM25 (Postgres FTS)0,2390,3760,3220,471
ST MiniLM-L6-v20,2250,3710,2870,426
ST BioBERT0,1930,3140,2520,373
ST MPNet-v20,1610,3170,2260,376
F1 und MAP@10 auf CodiESP auf Ebene des exakten Codes und der Kategorie.Tietai Cross-EncoderTietai Bi-EncoderBM25 (Postgres FTS)ST MiniLM-L6-v2ST BioBERTST MPNet-v20%20%40%60%80%100%F1 exaktF1 KategorieMAP@10 exaktMAP@10 Kategorie
F1 und MAP@10 auf CodiESP. Der Tietai-Cross-Encoder führt bei jeder Metrik. MAP@10 liegt über F1, was zeigt, dass viele Top-1-Fehler den richtigen Code dennoch in der Top-10-Auswahl enthalten.

Top-1-Precision, -Recall und -F1 für den exakten Code auf CodiESP

ModellPrecisionRecallF1
Tietai Cross-Encoder0,7090,7090,709
Tietai Bi-Encoder0,3590,3590,359
BM25 (Postgres FTS)0,2540,2270,239
ST MiniLM-L6-v20,2250,2250,225
ST BioBERT0,1930,1930,193
ST MPNet-v20,1610,1610,161
Top-1-Precision, -Recall und -F1 auf CodiESP auf Ebene des exakten Codes.Tietai Cross-EncoderTietai Bi-EncoderBM25 (Postgres FTS)ST MiniLM-L6-v2ST BioBERTST MPNet-v20%20%40%60%80%100%PrecisionRecallF1
Top-1-Precision, -Recall und -F1 für den exakten Code auf CodiESP. Dense Retriever geben eine Vorhersage pro Anfrage aus, daher fallen die drei Metriken zusammen. BM25 weicht ab, weil es für etwa 11 % der CodiESP-Anfragen keinen Kandidaten liefert.
Precision@k auf CodiESP nach Retrieval-Modell, für Treffer auf Ebene des exakten Codes und der Kategorie.Tietai Cross-EncoderTietai Bi-EncoderBM25 (Postgres FTS)ST MiniLM-L6-v2ST BioBERTST MPNet-v2Exakter Code0%20%40%60%80%100%13510kKategorie (3 Zeichen)13510k
Precision@k auf CodiESP. Die Precision sinkt mit k, weil jede Anfrage meist nur einen Gold-Code hat, aber die Reihenfolge der Modelle ist auf Ebene des exakten Codes und der Kategorie stabil.
Recall@k auf CodiESP nach Retrieval-Modell, für Treffer auf Ebene des exakten Codes und der Kategorie.Tietai Cross-EncoderTietai Bi-EncoderBM25 (Postgres FTS)ST MiniLM-L6-v2ST BioBERTST MPNet-v2Exakter Code0%20%40%60%80%100%13510kKategorie (3 Zeichen)13510k
Recall@k auf CodiESP. Der Cross-Encoder erreicht R@10 = 0,813 exakt und 0,903 auf Kategorieebene, während der Bi-Encoder in tieferen Rängen einen Großteil des Abstands aufholt.

Top-1-Genauigkeit nach Hierarchie auf CodiESP

Top-1-Genauigkeit auf CodiESP auf Ebene von exaktem Code, Kategorie und Kapitel.Tietai Cross-EncoderTietai Bi-EncoderBM25 (Postgres FTS)ST MiniLM-L6-v2ST BioBERTST MPNet-v20%20%40%60%80%100%Exakter CodeKategorie (3 Zeichen)Kapitel
Hierarchische Genauigkeit auf CodiESP. Dieselbe Top-1-Entscheidung verbessert sich vom exakten Code zur Kategorie und zum Kapitel – ein Hinweis darauf, wie oft Modelle die richtige Krankheitsfamilie finden, selbst wenn der vollständige Code falsch ist.

F1 und MAP@10 auf DISTEMIST

ModellF1 exaktF1 KategorieMAP@10 exaktMAP@10 Kategorie
Tietai Cross-Encoder0,7760,8180,8120,846
Tietai Bi-Encoder0,6030,6900,6820,747
BM25 (Postgres FTS)0,4310,4700,5040,537
F1 und MAP@10 auf DISTEMIST auf Ebene des exakten Codes und der Kategorie.Tietai Cross-EncoderTietai Bi-EncoderBM25 (Postgres FTS)0%20%40%60%80%100%F1 exaktF1 KategorieMAP@10 exaktMAP@10 Kategorie
F1 und MAP@10 auf DISTEMIST. Die relative Reihenfolge von Cross-Encoder, Bi-Encoder und BM25 bleibt auf einem unabhängigen spanischen klinischen Korpus unverändert.
Precision@k auf DISTEMIST nach Retrieval-Modell, für Treffer auf Ebene des exakten Codes und der Kategorie.Tietai Cross-EncoderTietai Bi-EncoderBM25 (Postgres FTS)Exakter Code0%20%40%60%80%100%13510kKategorie (3 Zeichen)13510k
Precision@k auf DISTEMIST. Die Precision sinkt mit wachsendem k, aber die aufgabenspezifischen Dense-Modelle bleiben auf Ebene des exakten Codes und der Kategorie vor der lexikalischen BM25-Baseline.
Top-1-Genauigkeit auf DISTEMIST auf Ebene von exaktem Code, Kategorie und Kapitel.Tietai Cross-EncoderTietai Bi-EncoderBM25 (Postgres FTS)0%20%40%60%80%100%Exakter CodeKategorie (3 Zeichen)Kapitel
Hierarchische Genauigkeit auf DISTEMIST. Die Genauigkeit steigt vom exakten Code zur Kategorie und zum Kapitel, während die Reihenfolge der Modelle gleich bleibt: Cross-Encoder, Bi-Encoder, dann BM25.

04 - Interpretation

Der Engpass ist die Aufgabenausrichtung, nicht die Embedding-Größe.

Die Reihenfolge der öffentlichen Baselines ist das wichtige Negativergebnis. BioBERT-ST hat biomedizinisches Pretraining, ist aber englischzentriert. MPNet-v2 hat größere Embeddings, liegt aber trotzdem hinter MiniLM-L6-v2. Weder Größe noch generisches Domänen-Pretraining lösen das Retrieval spanischer klinischer Codes allein.

ICD-Retrieval verlangt, dass Modelle Geschwisterkonzepte trennen, die semantisch nah, aber operativ verschieden sind. Der Bi-Encoder erfasst die thematische Information auf Kapitelebene gut, aber das Ranking des exakten Codes braucht einen Reranker, der mit Hard Negatives aus benachbarten CIE-10-Codes trainiert wurde.

  • Aufgabenausrichtung zählt. Synthetische Beispiele müssen die ICD-Hierarchie widerspiegeln, nicht nur eine breite biomedizinische Verwandtschaft.
  • Hard Negatives zählen. Benachbarte Codes bringen dem Modell bei, Ätiologie, Anatomie, Schweregrad und Episodentyp zu unterscheiden.
  • Reranking zählt. Der Cross-Encoder fügt die Unterscheidung auf Qualifikatorebene hinzu, die die CIE-10-ES-Richtlinien verlangen.
  • Wenig Daten können genügen. Ein kompakter synthetischer Korpus kann einen konkurrenzfähigen Retriever trainieren, wenn die Beispiele auf die Struktur des Zielvokabulars abgestimmt sind.
  • Biomedizinisches Pretraining überträgt sich nicht automatisch. Englische biomedizinische Encoder haben weiterhin Mühe, wenn sowohl Anfrage als auch Zielvokabular spanisch sind.

Für die produktive Suche in der klinischen Kodierung lautet die praktische Lehre: für das Zielvokabular, die Zielsprache und das Ziel-Ranking-Problem trainieren. Die nützliche Einheit ist kein größeres Embedding-Modell, sondern ein Retriever, der auf die Unterscheidungen trainiert ist, die Kodierfachkräfte tatsächlich treffen müssen.

Vollständige Arbeit lesen

Freie Bahn für Gesundheit.