Check if it makes sense to implement ViewModel support on top of RetainedInstanceStore. Similarly to this, one of the possible solutions:
Node implements ViewModelStoreOwner interface
RetainedInstanceStore stores and clears ViewModelStore for every Node