Revert checkbox change under treeTable in Primefaces

My application has a treeTable under Primfaces 6.1.1. Currently the application is using RequestContext.getCurrentInstance().update("myFormId:myTabViewId:myTreeTableId") to update the whole treeTable when the events select or unselect are triggered by ajax.

Due to performance issue when the treeTable contains so many nodes, I would like to update just the nodes that are affected by unselecting and selecting. Through searching, I found that there is no way to update just a part of treeTable, so I decided to update the treeTable-UI cell by cell by their Ids. This workaround works fine but I have the following problem:

when the user unselects a checkbox, a dialog will be shown to ask for confirmation, if the user clicks "No", the dialog will disappear and the application shouldn't do anything BUT the checkbox is not reverted back to the state "checked" although the internal CheckboxTreeNode.selected is still true.

I couldn't manage to update the checkbox in the UI by this workaround because the checkboxes are icons having no Ids and they are not found in the .xhml file.

I would like to ask if there is any way to update the checkbox-UI one by one on treeTable? Thanks.