PETRA: Transforming Web Text for Petroleum-Engineering Domain Adaptation

Kirill Dubovikov, Omar El Mansouri, Hachem Madmoun, Yanda Li, Sandeep Kumar +10 more
6/23/2026
cs.IRcs.CL

No confident code implementation found for this paper.

Abstract

Petroleum-engineering search exposes a supervision gap for strong general retrievers: relevant evidence exists in public web text, but domain relevance labels are scarce. To address this gap, we propose PETRA, a large-scale Petroleum Engineering Text for Retrieval Adaptation dataset and pipeline that converts noisy public web data into a curated domain corpus and synthetic supervision for dense retrieval and reranking. PETRA contains 1.36M curated chunks, approximately 2B token equivalents, $\approx$859k, embedding training rows from $\approx$224k anchors, and roughly 400k teacher-scored reranker candidate rows. Its construction combines high-recall energy-domain curation, an energy-domain classifier with 98.4% test accuracy, chunk-grounded query generation, LLM-written hard negatives, and retrieval-mined candidate lists. PETRA improves first-stage in-domain Normalized Discounted Cumulative Gain (nDCG) from 0.703 to 0.763 through score fusion. Reranker adaptation improves the public Earth Science benchmark by 44% relative and a six-task reasoning-intensive panel by 23%. Failed training recipes show that high train-holdout accuracy on synthetic labels does not predict retrieval gains; retrieval-mined data helps only after being repackaged as teacher-scored candidate lists sampled from the inference-time candidate distribution.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{dubovikov2026petra,
  title  = {PETRA: Transforming Web Text for Petroleum-Engineering Domain Adaptation},
  author = {Kirill Dubovikov and Omar El Mansouri and Hachem Madmoun and Yanda Li and Sandeep Kumar and Aya El Mir and Supriyo Ghosh and Writabrata Bhattacharya and Adrian Garcia-Garcia and Onkar Pandit and Sunil Kumar Sahu and Federico Castanedo and Larry Murray and Martin Takac and Salem Lahlou},
  year   = {2026},
  eprint = {2606.24346},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2606.24346v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2606.24346.svg)](https://codeofpaper.com/paper/2606.24346)

Discussion