|
Improving Object-centric Learning With Query Optimization
Baoxiong Jia*,
Yu Liu*,
Siyuan Huang,
ICLR 2023
[Paper]
 
[Project Page]
 
[Code]
We proposed methods for (1) initializing Slot-Attention modules with learnable queries and
(2) optimizing the model with bi-level optimization. With simple code adjustments on the
vanilla Slot-Attention, our model, Bi-level Optimized Query Slot Attention,
achieves state-of-the-art results on 3 challenging synthetic and 7 complex real-world datasets
in unsupervised image segmentation and reconstruction, outperforming previous baselines by a large margin.
|
 |
Beijing Institute for General Artificial Intelligence(BIGAI), China
2021.09 - now
Research Intern
Advisor: Dr. Baoxiong Jia
|
 |
Tsinghua University, China
2020.08 - now
Undergraduate Student
|
|