Interpretable DNA Sequence Classification via Dynamic Feature Generation in Decision Trees
Organizations: University of Cambridge · CU Anschutz Medical Campus
Abstract
The analysis of DNA sequences has become critical in numerous fields, from evolutionary biology to understanding gene regulation and disease mechanisms. While deep neural networks can achieve remarkable predictive performance, they typically operate as black boxes. Contrasting these black boxes, axis-aligned decision trees offer a promising direction for interpretable DNA sequence analysis, yet they suffer from a fundamental limitation: considering individual raw features in isolation at each split limits their expressivity, which results in prohibitive tree depths that hinder both interpretability and generalization performance. We address this challenge by introducing DEFT, a novel framework that adaptively generates high-level sequence features during tree construction. DEFT leverages large language models to propose biologically-informed features tailored to the local sequence distributions at each node and to iteratively refine them with a reflection mechanism. Empirically, we demonstrate that DEFT discovers human-interpretable and highly predictive sequence features across a diverse range of genomic tasks.
Figures & tables
| Pol II Pausing | Promoters | Enhancers | ||||
|---|---|---|---|---|---|---|
| Method | Acc. | AUPRC | Acc. | AUPRC | Acc. | AUPRC |
| CNN | ||||||
| Transformer | ||||||
| XGBoost | ||||||
| Log. Reg. | ||||||
| CART | ||||||
| Train | Test | |||
|---|---|---|---|---|
| DEFT | DEFT | |||
| 1 | ||||
| 2 | ||||
| 3 | ||||
| Model | Hal. Vol. | Hal. Effort | Hal. Dif. |
|---|---|---|---|
| DEFT | 15.5 | 15.5 | 0.875 |
| 19.7 | 17.3 | 1.0 |
Appendix figures & tables18 assets
Supplementary material from the paper’s appendix.
Appendix
| Characteristic | Value |
|---|---|
| Total number of samples | |
| Training set size | |
| Test set size | |
| Dimensionality of the sequences | |
| Label distribution (training set) | Pausing: , Non-pausing: |
| Characteristic | Value |
|---|---|
| Total number of samples | |
| Training set size | |
| Test set size | |
| Dimensionality of the sequences | |
| Label distribution (training set) | 1: , 0: |
| Characteristic | Value |
|---|---|
| Total number of samples | |
| Training set size | |
| Test set size | |
| Dimensionality of the sequences | |
| Label distribution | 1: , 0: |
| Parameter | Value |
|---|---|
| Model name | gpt-4o |
| Version | 1001 |
| Temperature | |
| Top p |
| Hyperparameter | Values |
|---|---|
| Conv1 filters | {32, 64} |
| Conv1 kernel size | {2, 12, 16} |
| Conv2 filters | {64, 128} |
| Dense units | {64, 128} |
| Dropout rate | {0, 0.2, 0.5} |
| Learning rate |
| Hyperparameter | Values |
|---|---|
| Hidden dimension | {32, 64, 128} |
| Attention heads | {2, 4} |
| Encoder layers | {1, 2, 4} |
| Dropout rate | {0, 0.2, 0.5} |
| Learning rate | |
| Batch size | {32, 128} |
| Section / Reference | Summary | Results |
|---|---|---|
| Section C.1 | Feature found without interpretability constraints. | — |
| Section C.2 | Unregularized CART increasingly overfits as depth grows. | Fig. 6 |
| Section C.3 | F1, precision, recall, and AUPRC across datasets. | Fig. 7 |
| Section C.4 | DEFT vs. Nucleotide Transformer (fine tuned with LoRA) for context. | Tab. 12 |
| Section C.5 | Specialized CNN/attention baselines: PEPMAN (Pol II), DeePromoter (Promoters). | Tab. 13 , Tab. 14 |
| Section C.6 | DEFT instantiated with an open-source LLM ( gpt-oss ). | Tab. 15 , |
| Hyperparameter | Value |
|---|---|
| LoRA rank ( ) | 1 |
| LoRA alpha ( ) | 32 |
| LoRA dropout | 0.1 |
| Optimizer | AdamW |
| Learning rate | |
| Train batch size | 8 |
| Pol II Pausing | Promoters | Enhancers | ||||
|---|---|---|---|---|---|---|
| Method | Acc. | AUPRC | Acc. | AUPRC | Acc. | AUPRC |
| CART ( -mers) | ||||||
| DEFT | ||||||
| Nucleotide Transformer | ||||||
| Model | Accuracy | AUPRC |
|---|---|---|
| CART (2-mers) | ||
| DEFT | ||
| PEPMAN |
| Model | Accuracy | AUPRC |
|---|---|---|
| CART (2-mers) | ||
| DEFT | ||
| DeePromoter |
| Pol II Pausing | Promoters | Enhancers | ||||
|---|---|---|---|---|---|---|
| Method | Acc. | AUPRC | Acc. | AUPRC | Acc. | AUPRC |
| CART ( -mers) | ||||||
| DEFT ( gpt-4o ) | ||||||
| DEFT ( gpt-oss ) | ||||||
| Parameter | Value |
|---|---|
| Model name | gpt-oss 120b |
| Temperature | |
| Top p | |
| Reasoning effort | Medium |
| Pol II Pausing | Promoters | Enhancers | ||||
|---|---|---|---|---|---|---|
| Method | Acc. | AUPRC | Acc. | AUPRC | Acc. | AUPRC |
| DEFT | 0.870 | 0.917 | 0.830 | 0.926 | 0.713 | 0.943 |
| DEFT (Ensemble) | 0.884 | 0.937 | 0.848 | 0.942 | 0.809 | 0.969 |
| Pol II Pausing | Promoters | Enhancers | ||||
|---|---|---|---|---|---|---|
| Method | Acc. | AUPRC | Acc. | AUPRC | Acc. | AUPRC |
| DNABERT-2 | ||||||
| HyenaDNA | ||||||
| CART | ||||||
| DEFT | ||||||
Interpretable ML
Interpretable machine learning develops models and explanations that make predictions, evidence, and decision processes understandable to people.
See all 306 papers