If you often find you want to go back more than the default segmentation undo history and you have enough memory in your computer to store more states then you can increase the maximum number of undo states.
However, it might be better to just save your work more frequently. You can easily save independent copies of the full scene by clicking on the package icon in Save data window; or you can quickly save just the modified nodes by saving using the default options.