The Effect of Document Selection on Query-focused Text Analysis
Organizations: Johns Hopkins University
Abstract
Analyses of document collections often require selecting what data to analyze, as not all documents are relevant to a particular research question and computational constraints preclude analyzing all documents, yet little work has examined effects of selection strategy choices. We systematically evaluate seven selection methods (from random selection to hybrid retrieval) on outputs from four text analyses methods (LDA, BERTopic, TopicGPT, HiCode) over two datasets with 26 open-ended queries. Our evaluation reveals practice guidance: semantic or hybrid retrieval offer strong go-to approaches that avoid the pitfalls of weaker selection strategies and the unnecessary compute overhead of more complicated ones. Overall, our evaluation framework establishes data selection as a methodological decision, rather than a practical necessity, inviting the development of new strategies.
Figures & tables
| Method | TREC-COVID | Doctor-Reviews | ||||||
|---|---|---|---|---|---|---|---|---|
| TopicGPT | BERTopic | LDA | HiCode | TopicGPT | BERTopic | LDA | HiCode | |
| Random Uniform | .156 ±.035 | .232 ±.030 | .266 ±.043 | .648 ±.115 | .224 ±.077 | .293 ±.079 | .397 ±.106 | .580 ±.051 |
| Keyword Search | .326 ±.072 | .325 ±.066 | .575 ±.082 | .679 ±.105 | .341 ±.106 | .404 ±.071 | .543 ±.082 | .628 ±.033 |
| SBERT | .398 ±.078 | .458 ±.028 | .596 ±.065 | .704 ±.083 | .402 ±.097 | .446 ±.055 | .501 ±.081 | .627 ±.044 |
| Direct Retrieval | .435 ±.135 | .407 ±.038 | .606 ±.071 | .676 ±.092 | .324 ±.116 | .471 ±.120 | .577 ±.076 | .624 ±.042 |
| Dir. Retr. + MMR | .265 ±.065 | .436 ±.119 | .594 ±.044 | .680 ±.088 | .288 ±.055 | .397 ±.064 | .513 ±.101 | .619 ±.045 |
Appendix figures & tables12 assets
Supplementary material from the paper’s appendix.
Appendix
| Topic ID | Docs | Description & Top Words |
| Violence-Related Topics | ||
| T2* | 270 | Victimization : violence, domestic violence, mental health, women |
| T0* | 27 | Crime and Public Safety : crime, impact, cities, pandemics |
| T3* | 35 | Social Compliance : social distancing, behaviour, social norms |
| T6* | 6 | Cybercrime : cybercrime, internet, attacks, inflicted violence |
| Non-Violence Topics | ||
| Topic ID | Docs | Description & Top Words |
| Violence-Related Topics | ||
| T1* | 145 | Family Violence : violence, domestic violence, women, IPV |
| T0* | 29 | Crime and Justice : crime, justice, police, burglary, violence |
| T4* | 11 | Cybercrime : cybercrime, crimes, conspiracy, fraud, violence |
| T5* | 3 | Gun Violence : gun violence, firearms, Los Angeles, Chicago |
| Non-Violence Topics | ||
| Selection | TopicGPT | BERTopic | LDA | HiCode |
|---|---|---|---|---|
| Random Uniform | 0.198 | 0.312 | 0.389 | 0.521 |
| Keyword Search | 0.267 | 0.401 | 0.421 | 0.558 |
| SBERT | 0.289 | 0.423 | 0.438 | 0.567 |
| Direct Retrieval | 0.304 | 0.448 | 0.447 | 0.573 |
| DR + MMR | 0.291 | 0.415 | 0.429 | 0.564 |
| Query Expansion | 0.298 | 0.431 | 0.441 | 0.569 |
| Topic ID | Docs | Description & Top Words |
| Violence-Related Topics (7 relevant) | ||
| T0* | 48 | Domestic Violence : domestic, violence, domestic violence, abuse, family |
| T9* | 45 | Violence Against Women : violence, women, IPV, intimate, partner |
| T2* | 15 | Crime Trends : crime, prison, burglary, prisons, frequency |
| T3* | 13 | Violence Against Health Workers : workers, kind warfare, healthcare workers, violence, law |
| T1* | 5 | Gun Violence : gun, guns, firearm, firearms, gun violence |
| Topic ID | Docs | Description & Top Words |
| Violence-Related Topics (5 relevant) | ||
| T2* | 106 | Intimate Partner Violence : violence, domestic violence, IPV, women |
| T0* | 36 | Crime and Safety : crime, violence, police, burglary |
| T11* | 16 | Police Violence : violence, women, police violence, law |
| T5* | 10 | Cybercrime : cybercrime, cyber, internet, violence |
| T29* | 6 | Race and Violence : race, violence, African Americans, racism |
| ID | Query Text | TopicGPT | BERTopic | LDA | HiCode | Status |
|---|---|---|---|---|---|---|
| 6 | Patients’ dislikes about doctors | 0.602 | 0.571 | 0.611 | 0.627 | Strong |
| 2 | Specialist referral experiences | 0.493 | 0.498 | 0.684 | 0.630 | Strong |
| 7 | Asthma follow-up care recommendations | 0.314 | 0.771 | 0.591 | 0.701 | Strong |
| 4 | Asthma patient management | 0.368 | 0.566 | 0.628 | 0.703 | Moderate |
| 8 | Likes about treatment recommendations | 0.528 | 0.486 | 0.592 | 0.635 | Moderate |
| 1 | Finding and choosing doctors | 0.462 | 0.457 | 0.648 | 0.649 | Moderate |
Data Selection
Data selection chooses informative, representative, or reliable examples to improve model training and evaluation while reducing data, compute, or labeling costs.
See all 82 papers