PhyloCNN: Improving tree representation and neural network architecture for deep learning from trees in phylodynamics and diversification studies

作者：Manolo F. Perez, Olivier Gascuel · 发表于：bioRxiv (Cold Spring Harbor Laboratory) · 年份：2024 · DOI：10.1101/2024.12.13.628187 · 被引用次数：6 · 研究领域：Evolution and Paleontology Studies、Genomics and Phylogenetic Studies、Genetic diversity and population structure

Abstract Phylodynamics and diversification studies using complex evolutionary models can be challenging, especially with traditional likelihood-based approaches. As an alternative, likelihood-free simulation-based approaches have been proposed due to their ability to incorporate complex models and scenarios. Here, we propose a new simulation-based deep learning (DL) method capable of analyzing large datasets and accurately estimating parameter values for birth-death models in both phylodynamics and diversification studies. Our approach involves encoding trees by extracting a vector of local features for all nodes of the input phylogeny. We also developed a dedicated convolutional neural network architecture called PhyloCNN. Using simulations, we compared the accuracy of PhyloCNN when using feature vectors with a variable number of generations to describe the local context of nodes and leaves. The number of generations had a greater impact when considering smaller training sets, with a broader context showing higher accuracy, especially for complex evolutionary models. Compared to other recently developed DL approaches, PhyloCNN showed higher or similar accuracies for all parameters when used with training sets one or two orders of magnitude smaller (10,000 to 100,000 simulated training trees, instead of millions). We applied PhyloCNN with compelling results to two real-world phylodynamics and diversification datasets, related to HIV superspreaders in Zurich and to primates an...