An Efficient Memory Module for Graph Few-Shot Class-Incremental Learning

Incremental graph learning has gained significant attention for its ability to address the catastrophic forgetting problem in graph representation learning. However, traditional methods often rely on a large number of labels for node classification, which is impractical in real-world applications. This makes few-shot incremental learning on graphs a pressing need. Current methods typically require extensive training samples from meta-learning to build memory and perform intensive fine-tuning of GNN parameters, leading to high memory consumption and potential loss of previously learned knowledge. To tackle these challenges, we introduce Mecoin, an efficient method for building and maintaining memory. Mecoin employs Structured Memory Units to cache prototypes of learned categories, as well as Memory Construction Modules to update these prototypes for new categories through interactions between the nodes and the cached prototypes. Additionally, we have designed a Memory Representation Adaptation Module to store probabilities associated with each class prototype, reducing the need for parameter fine-tuning and lowering the forgetting rate. When a sample matches its corresponding class prototype, the relevant probabilities are retrieved from the MRaM. Knowledge is then distilled back into the GNN through a Graph Knowledge Distillation Module, preserving the model's memory. We analyze the effectiveness of Mecoin in terms of generalization error and explore the impact of different distillation strategies on model performance through experiments and VC-dimension analysis. Compared to other related works, Mecoin shows superior performance in accuracy and forgetting rate. Our code is publicly available on the https://github.com/Arvin0313/Mecoin-GFSCIL.git .

Paper

Similar papers

Peer review

Reviewer utSJ6/10 · confidence 4/52024-07-01

Summary

This paper tackles the problem of graph few-shot class-incremental learning, proposes a novel framework named Mecoin to address the challenges of catastrophic forgetting and overfitting. Mecoin includes two key componets: the structured memory unit (SMU) for storing and updating prototypes and the memory representation adaptive module (MRaM) for separating the learning of prototypes and class representations. Based on SMU, Mecoin efficiently maintain representative prototypes preventing the overfitting problem caused by few-shot setting, and rely on MRaM, Mecoin avoids heavily finetuning the model parameters which solved the catastrophic forgetting problem in class-incremental learning.

Strengths

1.Stores past knowledge efficiently with low storage space by using the SMU. 2.Avoids catastrophic forgetting by separating the learning of prototypes and class representations. 3.The experiments show significant improvement compared to other state-of-the-art methods.

Weaknesses

1.The abbreviations of the proposed method are confusing. For example, in some places, the Memory Representation Adaptive Module is abbreviated as MRaM, while in others, it is MRAM. Additionally, the full model name, Mecoin, is too similar to one of its submodules, the Memory Construction module (MeCo). 2.Fig. 1 is confusing; it doesn’t show the details and advantages of the SMU and MRaM structures. 3.In Sec. 3.1, Eq. 6 lacks explanatory information. How does Eq. 6 integrate edge information into the loss function? 4.The proof of Theorem 1 is too brief. I recommend the authors provide detailed derivation rather than citing a lemma from another paper. 5.In Sec. 3.2, Eqs. 8 and 9 are incomprehensible. I assume the GKIM maintains past knowledge and updates new knowledge with different elements 𝑝. For example, if the dataset has 100 classes, 𝑝1:60 would be maintained for past knowledge and 𝑝61:100 for new knowledge. Thus, 𝑁 should be 60. It’s challenging to explain how new knowledge is updated in Eq. 9. 6.In Sec. 4.1, the process of pretraining the GNN needs to be explained. Is the pretraining phase a standard graph node classification task or a few-shot node classification task? Additionally, in Tab. 3, the comparison method MAG-Meta is mentioned twice; I guess one should be Geometer. Also, if the proposed method is named Mecoin, why does Mecoin appear among the comparison methods in Tabs. 2, 3, and 4? 7.In Sec. 4.2, the toy dataset is too simple. Sampling only 4 classes makes it difficult to demonstrate the ability of different settings to handle overlapping problems. I recommend sampling at least 10 classes.

Questions

Please check the weaknesses

Rating

6

Confidence

4

Soundness

3

Presentation

2

Contribution

3

Limitations

The paper is poor-writing and many places lack detailed explanations.

Reviewer cvjp7/10 · confidence 3/52024-07-07

Summary

The authors focus on graph few-shot class-incremental learning. The authors first introduce Mecoin to efficiently construct and preserve memory. To avoid extensive parameter finetuning and forgetting, the authors introduce a memory representation adaptive module called MRaM to separate the learning of prototypes and class representations. Besides, the authors propose Graph Knowledge Interchange Module (GKIM) to injects past knowledge information into GNN. Additional analyses illustrate the effectiveness of the methods.

Strengths

1. The paper is well-written, and the motivations for each part are clear. 2. The reported performance out-performs many baselines. 3. The theoretical and experimental analyses successfully illustrate the effectiveness of the methods.

Weaknesses

1. The paper does not report error bars, standard deviations or provide detailed information about the statistical significance of the experiments, which is important for understanding the reliability and variability of the results.

Questions

See Weaknesses.

Rating

7

Confidence

3

Soundness

3

Presentation

3

Contribution

3

Limitations

The paper has a section discussing several limitations in the appendix.

Reviewer f8WD5/10 · confidence 3/52024-07-13

Summary

This paper mainly focuses on graph few-shot class-incremental learning. To alleviate the significant memory consumption and catastrophic forgetting of old knowledge, it proposes to store only old class prototypes and update class prototypes by considering the interaction among nodes and prototypes. Furthermore, it gives a theoretical analysis on the generalization error. Experimental results show that the proposed method can achieve better results.

Strengths

The proposed idea of storing old class prototypes and updating class prototypes by considering their interaction with node features seems reasonable, as it has been proved to be useful in few-shot class-incremental learning. Experiments are relative sufficient to show its efficacy.

Weaknesses

In general, this paper needs further polish, from my point of view. Just name a few, the abbreviations MRaM and MRAM are not consistent; the adjacency matrix A is incorrect; some sentences need careful attention. In addition, the motivation or the reason why the proposed method uses self-attention and MRAM is not clearly explained.

Questions

1.What are the meta-learning samples? There is no definition or explanation about them. It would be much better if more explanation would be given. 2.Why using Gaussian random projection to reduce the dimensionality, instead of others? 3.Why concatenating G_T and H_T^p? Could the authors give more explanation? 4.What is the difference between class prototype and class representation? It is very hard to tell from the paper.

Rating

5

Confidence

3

Soundness

2

Presentation

2

Contribution

2

Limitations

The authors have discussed the limitations in the paper.

Reviewer utSJ2024-08-10

Request for Detailed Accuracy Metrics to Evaluate Class Incremental Learning Performance

Dear Authors, Thank you for your thorough rebuttal, which has successfully addressed most of my concerns. However, I would like to raise an additional point regarding the evaluation of the few-shot class incremental learning approach presented in your paper. A well-recognized challenge in the field of class incremental learning is the model’s ability to maintain high accuracy for the initial session classes while often struggling with the classes introduced in subsequent sessions. The metrics you have provided give an overall picture of the model’s performance but do not sufficiently highlight this specific issue. To better assess the model’s capability in this regard, I kindly ask that you report the accuracy metrics in a more granular fashion. Specifically, I would like to see the accuracy for the classes of each session reported separately. For instance, at session 2, it would be beneficial to have the following: The accuracy for the classes introduced in sessions 0 and 1. The accuracy for the classes introduced in session 2. This division will allow us to more clearly evaluate the model’s performance on both the older and the new classes, providing a more comprehensive understanding of its incremental learning capabilities. Thank you for considering this request, and I look forward to your response.

Authorsrebuttal2024-08-11

Thank you very much for your thorough review of our paper and for acknowledging our response. Your valuable feedback aids in a more comprehensive and clearer evaluation of our model. Since our experimental setup is consistent with previous papers [1][2], we did not report the specific accuracy for new and old classes in each session in detail. We fully agree that the evaluation method is crucial for a comprehensive understanding of our model's incremental learning capability. Based on your suggestion, we will add class accuracy metrics for each session in the results section. We apologize that due to word limits, we cannot present all experimental results (three tables, with a total of 153 columns and 19 rows each). In incremental learning tasks, as the number of sessions increases, the model’s ability to remember previous knowledge typically declines. Therefore, the newer sessions are better for evaluating the model's retention of past knowledge. Consequently, we have chosen to showcase the results of the last session for each dataset. The results are as follows: |methods|backbone|dataset||||||CoraFull|||||| |-|-|-|-|-|-|-|-|-|-|-|-|-|-| | \ | \ | \ ||||||session10|||||| | \ | \ | \ |session0|session1|session2|session3|session4|session5|session6|session7|session8|session9|session10| |ergnn|GCN|CoraFull|16.47|9.45|3.65|7.72|5.29|8.85|3.57|4.93|6.44|6.34|4.47| ||GAT|CoraFull|15.07|8.08|2.17|9.16|6.33|9.03|1.13|5.99|7.883|7.11|4.93| |lwf|GCN|CoraFull|9.15|6.45|4.86|5.58|6.24|6.39|9.22|9.81|2.88|7.81|1.49| ||GAT|CoraFull|9.15|6.45|4.86|5.58|6.24|6.39|9.22|9.81|2.88|7.81|1.49| |gem|GCN|CoraFull|9.23|7.11|8.93|5.87|6.39|6.65|9.48|9.95|3.92|7.57|2.95| ||GAT|CoraFull|8.52|7.17|8.61|3.58|6.25|6.78|8.89|9.76|2.96|6.12|2.38| |EWC|GCN|CoraFull|8.48|7.87|8.39|4.02|7.08|7.25|9.01|9.52|2.28|6.71|1.94| ||GAT|CoraFull|14.92|7.31|7.47|9.22|6.76|8.32|8.92|4.09|7.97|6.92|4.28| |MAS|GCN|CoraFull|15.75|8.38|9.47|9.52|6.27|8.37|9.17|4.48|7.63|7.74|4.6| ||GAT|CoraFull|45.96|40.11|41.69|22.09|19.25|17.76|17.21|16.38|10.41|11.51|12.34| |TWP|GCN|CoraFull|14.12|10.43|7.72|5.28|8.27|8.65|9.79|5.79|2.45|6.82|5.83| ||GAT|CoraFull|12.35|9.89|7.54|5.61|7.46|8.45|8.47|5.44|2.31|5.71|5.92| |Geometer| \ |CoraFull|9.9|9.79|8.84|6.01|8.87|8.98|8.52|5.12|3.33|4.13|4.58| |HAG-Meta| \ |CoraFull|60.74|50.18|45.16|32.86|21.43|24.42|16.95|24.21|11.45|20.68|13.28| |ours|GCN|CoraFull|76.52|36|58.23|24.18|21.43|33.51|7.63|23.12|12.25|32.23|14.09| ||GAT|CoraFull|76.52|44|58.35|32.009|36.54|22.68|34.73|19.89|5.5|36.74|20.88| |methods|backbone|dataset||||||CS|||||| |-|-|-|-|-|-|-|-|-|-|-|-|-|-| | \ | \ | \ ||||||session10|||||| | \ | \ | \ |session0|session1|session2|session3|session4|session5|session6|session7|session8|session9|session10| |ergnn|GCN|CS|0|0|0|0|0|0|0|0|0|24.39|100| ||GAT|CS|0|0|0|14.83|6.28|2.48|5.48|2.81|21.78|33.34|100| |lwf|GCN|CS|0|0|0|0|0|0|0|0|12.36|11.13|100| ||GAT|CS|0|0|0|17.76|13.6|7.02|5.34|6.15|21.61|31.87|100| |gem|GCN|CS|0|0|0|0|0|0|0|0|0|15.59|100| ||GAT|CS|0|0|0|0|0|7.55|0|0|0|15.35|100| |EWC|GCN|CS|0|0|0|0|0|0|0|0|0|17.12|100| ||GAT|CS|0|0|0|0|16.24|18.04|17.84|10.97|20.95|32.12|100| |MAS|GCN|CS|0|0|0|0|0|0|0|0|0|12.89|98.23| ||GAT|CS|0|0|41.43|41.17|35.06|52.37|48.78|31.29|25.53|38.02|100| |TWP|GCN|CS|0|0|0|0|0|0|0|0|0|26.97|100| ||GAT|CS|0|8.52|40.34|38.16|30.12|49.97|44.12|20.88|18.23|24.8|94.56| |Geometer| \ |CS|2.85|13.45|8.28|10.69|8.74|8.09|3.75|3.79|17.92|31.63|86.57| |HAG-Meta| \ |CS|9.89|8.91|7.31|2.54|5.69|3.73|7.53|6.41|4.87|5.98|4.65| |ours|GCN|CS|90.26|44.2|40.91|22.6|58.17|81.48|59.37|42.08|34.92|35.95|51.68| ||GAT|CS|91.51|47.28|43.1|23.84|61.43|80.11|63.79|45.33|37.47|39.74|54.92| |methods|backbone|dataset|||Computers|||| |-|-|-|-|-|-|-|-|-| | \ | \ | \ |||session5|||| | \ | \ | \ |session0|session1|session2|session3|session4|session5| |ergnn|GCN|Computers|0|0|0|0|0|100| ||GAT|Computers|0|0|0|0|0|100| |lwf|GCN|Computers|0|0|0|0|0|100| ||GAT|Computers|0|0|0|0|2.87|100| |gem|GCN|Computers|0|0|0|0|0|100| ||GAT|Computers|0|0|0|0|0|100| |EWC|GCN|Computers|0|0|0|0|0|100| ||GAT|Computers|0|0|0|0|0|100| |MAS|GCN|Computers|0|0|0|0|0|100| ||GAT|Computers|0|0|0|12.86|8.98|100| |TWP|GCN|Computers|0|0|0|0|0|100| ||GAT|Computers|0|0|0|0|0|100| |Geometer| \ |Computers|0|0|0|0|0|87.55| |HAG-Meta| \ |Computers|0|0|0|0|0|80.69| |ours|GCN|Computers|71.37|75.15|48.88|77.69|68.24|79.36| ||GAT|Computers|82.42|73.26|42.7|77.69|71.24|60.98| From the experimental results, it is evident that our method outperforms the baseline methods in both the older and the new classes, highlighting the advantages of our approach. If you would like to see results from any other session, we can provide them in a follow-up response. We will update the full experimental results in the appendix of the revised paper. [1] Graph few-shot class-incremental learning. [2] Geometer:Graph few-shot class-incremental learning via prototype representation.

Reviewer utSJ2024-08-11

I appreciate your considerate response to my latest concern regarding the detailed accuracy metrics for class incremental learning. The additional data and insights you have provided have sufficiently addressed my queries, and I am now convinced of the paper’s contributions to the field. From the result I can see the challenge of plasticity-stability dilemma still exists and needs to be overcome. As we move forward, I would like to suggest a final refinement to enhance the readability and coherence of the manuscript, especially, the method section. I kindly ask that you carefully rephrase the text to ensure that the paper is as clear and easy to understand as possible. This will not only benefit the peer-review process but also make the paper more accessible to a broader audience, including practitioners and researchers who may not be as deeply familiar with the subject matter. Your attention to detail in revising the manuscript will help to clarify the methodologies and findings, making the paper’s strengths even more apparent. I believe that with these adjustments, the paper will be well-positioned for acceptance and will have a significant impact on the community.

Authorsrebuttal2024-08-11

Thank you for further acknowledging and supporting our paper, as well as for your approval of our previous response. Your feedback is extremely valuable to us. We agree that the plasticity-stability dilemma remains a challenge to overcome, and we will continue to explore and address this issue in future research. Regarding your suggestion to further refine the manuscript, we will carefully consider and implement it. We plan to review the paper’s expression and logic, particularly in the methodology section, to ensure clarity and readability. We will improve the paper with the following measures: - **Optimize Paper Expression**: We will modify the descriptions of paragraphs and sections according to the paper’s logic, such as word choice and sentence structure, while appropriately adding relevant background introductions to make the expressions more complete, clear, and easy to understand. - **Improve Method Description**: We will enhance the introduction of the proposed method, Mecoin, to clarify the roles of each component and their interrelationships; refine the explanation of technical terms, and carefully check the standardized use of symbols. We will also articulate the theory and its conclusions more clearly to help readers better understand our method and contributions. Thank you once again for your invaluable suggestions. We believe that with these adjustments, our paper will better highlight its scientific contributions and be more accessible to a broader audience.

Authorsrebuttal2024-08-13

Dear Reviewer: We are honored by your thorough and detailed review. Your valuable suggestions have greatly enhanced our reflection on the paper and significantly improved the quality and depth of our research. Your insightful feedback has been instrumental in further elevating the paper's quality. Additionally, your recommendation to report accuracy metrics in a more detailed manner will aid us in better analyzing the plasticity-stability dilemma in continual learning. We particularly appreciate the professionalism and constructive feedback you have demonstrated during the rebuttal phase. Your suggestions have allowed us to improve the paper and ultimately receive your endorsement. This is both an encouragement and an inspiration to us, and we believe these improvements will make our work more appealing. Thank you again for your support and assistance. Sincerely.

Reviewer cvjp2024-08-12

Thanks for the Response

The authors response resolves my concerns, I decide to keep my rating.

Authorsrebuttal2024-08-13

Dear Reviewer: We greatly appreciate your detailed review and professional feedback on our paper. We especially thank you for pointing out the issue regarding the omission of error reporting in the experimental section. Based on your valuable comments, we have made the necessary improvements, updated the experimental report, and ensured the accuracy and completeness of the results. We are pleased to receive your positive evaluation, which is crucial for advancing this research direction. Your guidance has helped us enhance the quality of our paper and better meet academic standards. Thank you again for your support and assistance. Sincerely.

Reviewer f8WD2024-08-13

The rebuttal has addressed part of concerns.

After seeing the response from the authors, part of my concerns has been addressed. Since the rest of the reviewers liked the paper and also considering my low confidence, I tend to raise my rating to borderline accept. But I do encourage the authors to better polish their paper, especially the Method section, to make it easier to follow.

Authorsrebuttal2024-08-13

Dear Reviewer: We greatly appreciate your valuable feedback and suggestions during the review process. Your professional insights and thoughtful evaluation are highly significant to us. We are pleased to see your positive assessment of our paper. Based on your comments, we have made the following revisions: - **Refinement of Paper Expression**: We have re-examined our paper and made further modifications to the logic, sentence structure, and word choice. - **Standardization of Symbols and Abbreviations**: We carefully reviewed the symbols and abbreviations used in the paper and have standardized and unified them. - **Improved Method Description**: We have provided a more detailed introduction of the proposed method, Mecoin, clarifying the roles of each component and their interrelationships, and included more detailed explanations of the objectives and operations in the method section. We believe these changes will further enhance the quality of our paper. Thank you again for your support and assistance. Sincerely.

Program Chairsdecision2024-09-25

Decision

Accept (poster)

© 2026 NYSGPT2525 LLC