KS-GNN: Keyword Search via Graph Neural Network for Web API Recommendation

With the rapid development of service computing, a large number of methods for Web service recommendation have been proposed. However, the existing approaches using Mashup description information ignore the fact that the users without knowledge of Web APIs are not able to describe their needs in detail, let alone find Web services that meet those needs and are compatible with each other. Meanwhile, most approaches that utilize Web API collaboration network based on Mashup-API invocation relationships do not effectively capture the local and global structure between APIs and mine hidden API compatibility information in the network. This paper introduces the KS-GNN model, a novel approach that utilizes graph neural network and auto-encoder techniques for Web API recommendation. Firstly, we utilize KeyBert to extract keywords related to Web services from functional descriptions. Then, we embed the extracted keywords and use their embedded representations as node representation vectors on the Web API collaboration network. Finally, considering local and global structural relationships in the Web API collaborative network and the network structural relationships for message passing, KS-GNN performs keyword searching on the Web API collaborative network, to recommend the top-K Web services that match the user’s query. Experimental results on the ProgrammableWeb dataset show that KS-GNN outperforms other deep learning-based factorization machine recommendation models. In the meantime, we also confirm that the method of extracting keywords using KeyBert outperforms other keyword extraction methods.