DeepTaxon: An Interpretable Retrieval-Augmented Multimodal Framework for Unified Species Identification and Discovery
About
Identifying species in biology among tens of thousands of visually similar taxa while discovering unknown species in open-world environments remains a fundamental challenge in biodiversity research. Current methods treat identification and discovery as separate problems, with classification models assuming closed sets and discovery relying on threshold-based rejection. Here we present DeepTaxon, a retrieval-augmented multimodal framework that unifies species identification and discovery through interpretable reasoning over retrieved visual evidence. Given a query image, DeepTaxon retrieves the top-$k$ candidate species with $n$ exemplar images each from a retrieval index and performs chain-of-thought comparative reasoning. Critically, we redefine discovery as an explicit, retrieval-based decision problem rather than an implicit parametric memory problem. A sample is novel if and only if the retrieval index lacks sufficient evidence for identification, so each retrieval naturally yields a classification or discovery label without manual annotation, thereby providing automatic supervision for both tasks. We train the framework via supervised fine-tuning on synthetic retrieval-augmented data, followed by reinforcement learning on hard samples, converting high-recall retrieval into high-precision decisions that scale to massive taxonomic vocabularies. Extensive experiments on a large-scale in-distribution benchmark and six out-of-distribution datasets demonstrate consistent improvements in both identification and discovery. Ablation studies further reveal effective test-time scaling with candidate count $k$ and exemplar count $n$, strong zero-shot transfer to unseen domains, and consistent performance across retrieval encoders, establishing an interpretable solution for biodiversity research.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | FGVC-Aircraft (test) | Accuracy67.38 | 322 | |
| Image Classification | Stanford Cars (test) | Accuracy93.02 | 320 | |
| Image Classification | Stanford Dogs (test) | Top-1 Acc84.1 | 140 | |
| Image Classification | Food (test) | Accuracy89.98 | 128 | |
| Image Classification | Flowers102 (test) | Accuracy99.17 | 123 | |
| Image Classification | iNaturalist | Accuracy60.07 | 74 | |
| Error detection | Flowers102 | -- | 46 | |
| OOD Detection | Food101 | -- | 27 | |
| OOD Detection | FGVCAircraft | -- | 27 | |
| OOD Detection | Stanford Cars | -- | 21 |