79 citations · 103 across the 5 of their papers we have counts for
6 papers
A Flexible 2.5D Medical Image Segmentation Approach with In-Slice and Cross-Slice Attention
Amarjeet Kumar, Hongxu Jiang, Muhammad Imran +7
Deep learning has become the de facto method for medical image segmentation, with 3D segmentation models excelling in capturing complex 3D structures and 2D models offering high co…
Message Passing Neural Networks for Traffic Forecasting
Arian Prabowo, Hao Xue, Wei Shao +2
A road network, in the context of traffic forecasting, is typically modeled as a graph where the nodes are sensors that measure traffic metrics (such as speed) at that location. Tr…
Because Every Sensor Is Unique, so Is Every Pair: Handling Dynamicity in Traffic Forecasting
Arian Prabowo, Wei Shao, Hao Xue +2
Traffic forecasting is a critical task to extract values from cyber-physical infrastructures, which is the backbone of smart transportation. However owing to external contexts, the…
CLARE: Conservative Model-Based Reward Learning for Offline Inverse Reinforcement Learning
Sheng Yue, Guanbo Wang, Wei Shao +4
This work aims to tackle a major challenge in offline Inverse Reinforcement Learning (IRL), namely the reward extrapolation error, where the learned reward function may fail to exp…
Multiple-level Point Embedding for Solving Human Trajectory Imputation with Prediction
Kyle K. Qin, Yongli Ren, Wei Shao +3
Sparsity is a common issue in many trajectory datasets, including human mobility data. This issue frequently brings more difficulty to relevant learning tasks, such as trajectory i…
CoBEVT: Cooperative Bird's Eye View Semantic Segmentation with Sparse Transformers
Runsheng Xu, Zhengzhong Tu, Hao Xiang +3
Bird's eye view (BEV) semantic segmentation plays a crucial role in spatial sensing for autonomous driving. Although recent literature has made significant progress on BEV map unde…