Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Question about retraining #3

Open
Kangerlussuaq opened this issue Apr 23, 2024 · 1 comment
Open

Question about retraining #3

Kangerlussuaq opened this issue Apr 23, 2024 · 1 comment

Comments

@Kangerlussuaq
Copy link

Does this repository contain codes that handle network topology changes?
It is mentioned in section 4 of your paper that the model could be retrained when a node or link is added and it is faster than training from scratch.
Should I keep the parameters unchanged in nodes which are not affected by the add action, and initialize new parameters randomly?

@luopeng69131
Copy link

The repository does not seems to include the code dealing the topology changes about adding new link.
https://github.com/harvard-cns/teal/blob/main/lib/graph_utils.py#L9
In the link, you can see the function to add edge. But it is from the other repository. In this, it is not used.

In paper, the author also does not discuss more content that you're interested in.

About the retraining, due to only using linear parameters in FlowGNN, so I think you can use the old parameters to initialize the new FlowGNN and not need to adjust any architecture of the model.

Removing the edges or nodes, the model needs to be processed differently.
If the edge is removed, the paths and graph data of FlowGNN (variable: self.edge_index, self.edge_index_values, self.p2e in teal_env) must be recalculated. But the model and its action space doesn't need to change, because the number of node pairs doesn't change.
If the node is removed, the paths, graph data of FlowGNN and linear of output (in teal_actor init) need to be changed. The output action length is node_pairs * 4 (th number of path in every node pair).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants