NODEO: A Neural Ordinary Differential Equation Based Optimization Framework for Deformable Image Registration

Y. Wu, T. Z. Jiahao, J. Wang, P. A. Yushkevich, M. A. Hsieh, J. C. Gee

Abstract: Deformable image registration (DIR), aiming to find spatial correspondence between images, is one of the most critical problems in the domain of medical image analysis. In this paper, we present a novel, generic, and accurate diffeomorphic image registration framework that utilizes neural ordinary differential equations (NODEs). We model each voxel as a moving particle and consider the set of all voxels in a 3D image as a high-dimensional dynamical system whose trajectory determines the targeted deformation field. Our method leverages deep neural networks for their expressive power in modeling dynamical systems, and simultaneously optimizes for a dynamical system between the image pairs and the corresponding transformation. Our formulation allows various constraints to be imposed along the transformation to maintain desired regularities. Our experiment results show that our method outperforms the benchmarks under various metrics. Additionally, we demonstrate the feasibility to expand our framework to register multiple image sets using a unified form of transformation,which could possibly serve a wider range of applications.

Status: Accepted to CVPR 2022.

Preprint available on arXiv:

Topology Control of a Periodic Time-varying Communication Network with Stochastic Temporal Links

L. Shen, X. Yu, and M. A. Hsieh

Abstract: Mobile agents can form communication networks with links that emerge and disappear over time. Information transmission on such networks must pass through a sequence of links that are activated in a certain chronological order. Uncertainties in a link’s activation or deactivation risk violating the chronological order of the formation of links in paths for information transmission. The reasons for these uncertainties can be varied, but their presence prevents the application of existing topology control tools for mitigation. We propose an approach to leverage the pre-stored knowledge of the scheduled links’ events and create time-respecting subpaths accordingly. By estimating the impact of the stochastic timing on each subpath and by measuring how influential the subpaths are, we are able to measure the impact of uncertainties on any link event’s timing on the whole network’s information transmission performance. Reducing the uncertainty in the timing of those links with the highest impact can largely improve the network’s performance. Our method is tested and validated with simulation results.

Status: Accepted to ACC 2022.

Preprint available on arXiv: To come.

Flow-Based Control of Marine Robots in Gyre-Like Environments

G. Knizhnik, P. Li, X. Yu, and M. A. Hsieh

Abstract: We present a flow-based control strategy that enables resource-constrained marine robots to patrol gyre-like flow environments on an orbital trajectory with a periodicity in a given range. The controller does not require a detailed model of the flow field and relies only on the robot’s location relative to the center of the gyre. Instead of precisely tracking a pre-defined trajectory, the robots are tasked to stay in between two bounding trajectories with known periodicity. Furthermore, the proposed strategy leverages the surrounding flow field to minimize control effort. We prove that the proposed strategy enables robots to cycle in the flow satisfying the desired periodicity requirements. Our method is tested and validated both in simulation and in experiments using a low-cost, underactuated, surface swimming robot, i.e. the Modboat.

Status: Accepted to ICRA 2022.

Preprint available on arXiv:

Learning to Swarm with Knowledge-Based Neural Ordinary Differential Equations

T. Z. Jiahao, L. Pan, and M. A. Hsieh

Abstract: Understanding decentralized dynamics from collective behaviors in swarms is crucial for informing robot controller designs in artificial swarms and multiagent robotic systems. However, the complexity in agent-to-agent interactions and the decentralized nature of most swarms pose a significant challenge to the extraction of single-robot control laws from global behavior. In this work, we consider the important task of learning decentralized single-robot controllers based solely on the state observations of a swarm’s trajectory. We present a general framework by adopting knowledge-based neural ordinary differential equations (KNODE) — a hybrid machine learning method capable of combining artificial neural networks with known agent dynamics. Our approach distinguishes itself from most prior works in that we do not require action data for learning. We apply our framework to two different flocking swarms in 2D and 3D respectively, and demonstrate efficient training by leveraging the graphical structure of the swarms’ information network. We further show that the learnt single-robot controllers can not only reproduce flocking behavior in the original swarm but also scale to swarms with more robots.

Status: Accepted to ICRA 2022.

Preprint available on arXiv:

Cooperative Transport by ASVs

Differential Geometric Approach to Trajectory Planning: Cooperative Transport by a Team of Autonomous Marine Vehicles

Hadi Hajieghrary, Dhanushka Kularatne, and M. Ani Hsieh

Abstract: In this paper we addressed the cooperative transport problem for a team of autonomous surface vehicles (ASVs) towing a single buoyant load. We consider the dynamics of the constrained system and decompose the cooperative transport problem into a collection of subproblems. Each subproblem consists of an ASV and load pair where each ASV is attached to the load at the same point. Since the system states evolve on a smooth manifold, we use the tools from differential geometry to model the holonomic constraint arising from the cooperative transport problem and the non-holonomic constraints arising from the ASV dynamics. We then synthesize distributed feedback control strategies using the proposed mathematical modeling framework to enable the team transport the load on a desired trajectory. We experimentally validate the proposed strategy using a team of micro ASVs.

Status: To be presented at ACC 2018.  Preprint to come.

Path Planning with Forecast Uncertainties

Optimal Path Planning in Time-Varying Flows with Forecasting Uncertainties

Dhanushka Kularatne, Hadi Hajieghrary, and M. Ani Hsieh

Abstract: Uncertainties in flow models have to be explicitly considered for effective path planning in marine environments. In this paper, we present two methods to compute minimum expected cost policies and paths over an uncertain flow model. The first method based on a Markov Decision Process computes a minimum expected cost policy while the second graph search based method, computes a minimum expected cost path. A transition probability model is developed to compute the probability of transition from one state to another under a given action. In addition, a method to compute the expected cost of a path when it is executed in an uncertain flow field is also presented. The two methods are used to compute minimum energy paths in an ocean environment and the results are analyzed in simulations.

Status: To be presented at ICRA 2018 in Brisbane, Australia.  Preprint to come.