Tract-RLFormer: A Tract-Specific RL policy based Decoder-only Transformer Network

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Joshi, Ankita, Sharma, Ashutosh, Goel, Anoushkrit, Jha, Ranjeet Ranjan, Ahuja, Chirag, Bhavsar, Arnav, Nigam, Aditya
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915019041538048
author Joshi, Ankita
Sharma, Ashutosh
Goel, Anoushkrit
Jha, Ranjeet Ranjan
Ahuja, Chirag
Bhavsar, Arnav
Nigam, Aditya
author_facet Joshi, Ankita
Sharma, Ashutosh
Goel, Anoushkrit
Jha, Ranjeet Ranjan
Ahuja, Chirag
Bhavsar, Arnav
Nigam, Aditya
contents Fiber tractography is a cornerstone of neuroimaging, enabling the detailed mapping of the brain's white matter pathways through diffusion MRI. This is crucial for understanding brain connectivity and function, making it a valuable tool in neurological applications. Despite its importance, tractography faces challenges due to its complexity and susceptibility to false positives, misrepresenting vital pathways. To address these issues, recent strategies have shifted towards deep learning, utilizing supervised learning, which depends on precise ground truth, or reinforcement learning, which operates without it. In this work, we propose Tract-RLFormer, a network utilizing both supervised and reinforcement learning, in a two-stage policy refinement process that markedly improves the accuracy and generalizability across various data-sets. By employing a tract-specific approach, our network directly delineates the tracts of interest, bypassing the traditional segmentation process. Through rigorous validation on datasets such as TractoInferno, HCP, and ISMRM-2015, our methodology demonstrates a leap forward in tractography, showcasing its ability to accurately map the brain's white matter tracts.
format Preprint
id arxiv_https___arxiv_org_abs_2411_05757
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Tract-RLFormer: A Tract-Specific RL policy based Decoder-only Transformer Network
Joshi, Ankita
Sharma, Ashutosh
Goel, Anoushkrit
Jha, Ranjeet Ranjan
Ahuja, Chirag
Bhavsar, Arnav
Nigam, Aditya
Machine Learning
Fiber tractography is a cornerstone of neuroimaging, enabling the detailed mapping of the brain's white matter pathways through diffusion MRI. This is crucial for understanding brain connectivity and function, making it a valuable tool in neurological applications. Despite its importance, tractography faces challenges due to its complexity and susceptibility to false positives, misrepresenting vital pathways. To address these issues, recent strategies have shifted towards deep learning, utilizing supervised learning, which depends on precise ground truth, or reinforcement learning, which operates without it. In this work, we propose Tract-RLFormer, a network utilizing both supervised and reinforcement learning, in a two-stage policy refinement process that markedly improves the accuracy and generalizability across various data-sets. By employing a tract-specific approach, our network directly delineates the tracts of interest, bypassing the traditional segmentation process. Through rigorous validation on datasets such as TractoInferno, HCP, and ISMRM-2015, our methodology demonstrates a leap forward in tractography, showcasing its ability to accurately map the brain's white matter tracts.
title Tract-RLFormer: A Tract-Specific RL policy based Decoder-only Transformer Network
topic Machine Learning
url https://arxiv.org/abs/2411.05757