Hierarchical Sparse Bayesian Multitask Model with Scalable Inference for Microbiome Analysis
Fuente:
arXiv
Salvato in:
| Autori principali: | , , , , , , , , , , , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2025
|
| Soggetti: | |
| Accesso online: | |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866915137298890752 |
|---|---|
| author | Zhu, Haonan Goncalves, Andre R. Valdes, Camilo Ranganathan, Hiranmayi Zhang, Boya Martí, Jose Manuel Kok, Car Reen Borucki, Monica K. Mulakken, Nisha J. Thissen, James B. Jaing, Crystal Hero, Alfred Be, Nicholas A. |
| author_facet | Zhu, Haonan Goncalves, Andre R. Valdes, Camilo Ranganathan, Hiranmayi Zhang, Boya Martí, Jose Manuel Kok, Car Reen Borucki, Monica K. Mulakken, Nisha J. Thissen, James B. Jaing, Crystal Hero, Alfred Be, Nicholas A. |
| contents | This paper proposes a hierarchical Bayesian multitask learning model that is applicable to the general multi-task binary classification learning problem where the model assumes a shared sparsity structure across different tasks. We derive a computationally efficient inference algorithm based on variational inference to approximate the posterior distribution. We demonstrate the potential of the new approach on various synthetic datasets and for predicting human health status based on microbiome profile. Our analysis incorporates data pooled from multiple microbiome studies, along with a comprehensive comparison with other benchmark methods. Results in synthetic datasets show that the proposed approach has superior support recovery property when the underlying regression coefficients share a common sparsity structure across different tasks. Our experiments on microbiome classification demonstrate the utility of the method in extracting informative taxa while providing well-calibrated predictions with uncertainty quantification and achieving competitive performance in terms of prediction metrics. Notably, despite the heterogeneity of the pooled datasets (e.g., different experimental objectives, laboratory setups, sequencing equipment, patient demographics), our method delivers robust results. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2502_02552 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Hierarchical Sparse Bayesian Multitask Model with Scalable Inference for Microbiome Analysis Zhu, Haonan Goncalves, Andre R. Valdes, Camilo Ranganathan, Hiranmayi Zhang, Boya Martí, Jose Manuel Kok, Car Reen Borucki, Monica K. Mulakken, Nisha J. Thissen, James B. Jaing, Crystal Hero, Alfred Be, Nicholas A. Machine Learning Biomolecules Applications Computation Methodology This paper proposes a hierarchical Bayesian multitask learning model that is applicable to the general multi-task binary classification learning problem where the model assumes a shared sparsity structure across different tasks. We derive a computationally efficient inference algorithm based on variational inference to approximate the posterior distribution. We demonstrate the potential of the new approach on various synthetic datasets and for predicting human health status based on microbiome profile. Our analysis incorporates data pooled from multiple microbiome studies, along with a comprehensive comparison with other benchmark methods. Results in synthetic datasets show that the proposed approach has superior support recovery property when the underlying regression coefficients share a common sparsity structure across different tasks. Our experiments on microbiome classification demonstrate the utility of the method in extracting informative taxa while providing well-calibrated predictions with uncertainty quantification and achieving competitive performance in terms of prediction metrics. Notably, despite the heterogeneity of the pooled datasets (e.g., different experimental objectives, laboratory setups, sequencing equipment, patient demographics), our method delivers robust results. |
| title | Hierarchical Sparse Bayesian Multitask Model with Scalable Inference for Microbiome Analysis |
| topic | Machine Learning Biomolecules Applications Computation Methodology |
| url | https://arxiv.org/abs/2502.02552 |