A Primal-Dual Solver for Large-Scale Tracking-by-Assignment
About
We propose a fast approximate solver for the combinatorial problem known as tracking-by-assignment, which we apply to cell tracking. The latter plays a key role in discovery in many life sciences, especially in cell and developmental biology. So far, in the most general setting this problem was addressed by off-the-shelf solvers like Gurobi, whose run time and memory requirements rapidly grow with the size of the input. In contrast, for our method this growth is nearly linear. Our contribution consists of a new (1) decomposable compact representation of the problem; (2) dual block-coordinate ascent method for optimizing the decomposition-based dual; and (3) primal heuristics that reconstructs a feasible integer solution based on the dual information. Compared to solving the problem with Gurobi, we observe an up to~60~times speed-up, while reducing the memory footprint significantly. We demonstrate the efficacy of our method on real-world tracking problems.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Markov Random Field inference | MRF (C-seg) | Dual Objective Lower Bound2.00e+4 | 8 | |
| Cell tracking | Cell tracking Large | Dual Objective (Lower Bound)-1.55e+8 | 4 | |
| Cell tracking | Cell tracking Small | Dual Objective Bound-4.39e+6 | 4 | |
| Graph matching | Graph matching Hotel | Dual Objective (Lower Bound)-4.29e+3 | 4 | |
| Graph matching | Graph matching (House) | Dual Objective-3.78e+3 | 4 | |
| Markov Random Field inference | MRF Obj-seg | Dual Objective (Lower Bound)3.13e+4 | 4 | |
| Graph matching | Graph matching Worms | Dual Objective (Lower Bound)-4.85e+4 | 4 | |
| Markov Random Field inference | MRF (C-seg-n4) | Dual Objective Bound2.00e+4 | 4 |