I don't care for smartwatch features and pretty much only use it for sleep tracking and less so but also monitoring general heart health.
Given the lack of transparency on how FitBit (and others) infer sleep stages; it's hard to know for sure but I believe FitBit only uses heart rate data.
If that's the case, is it possible to feed data from a regular fitness bluetooth chest or arm heart rate strap into Fitbit (or similarly accurate) sleep stage inference systems?
Try https://github.com/ojwalch/sleep_classifiers and other work from Olivia Walch and her lab at UMichigan.
You may find other papers on sleep staging referenced at the National Sleep Research Resource https://sleepdata.org/ or at https://physionet.org/about/software/